transformers
9fd123ac - ci: mark model_parallel tests as cuda specific (#35269)

Comment changes are shownComment changes are hidden
Commit
162 days ago
ci: mark model_parallel tests as cuda specific (#35269) `parallelize()` API is deprecated in favor of accelerate's `device_map="auto"` and therefore is not accepting new features. At the same time `parallelize()` implementation is currently CUDA-specific. This commit marks respective ci tests with `@require_torch_gpu`. Fixes: #35252 Signed-off-by: Dmitry Rogozhkin <dmitry.v.rogozhkin@intel.com>
Author
Parents
  • tests
    • File
      test_modeling_common.py