Skip to content

GitLab

"benchmark/git@developer.sourcefind.cn:change/sglang.git" did not exist on "e17deb27b5cb3417c0315791ceddb5b4e40e7e91"
Unverified Commit fb5665b5 authored by Lysandre Debut's avatar Lysandre Debut Committed by GitHub
Browse files

The extended trainer tests should require torch (#12650)

parent 0af8579b
Show whitespace changes
Inline Side-by-side
Showing with 2 additions and 0 deletions
tests/extended/test_trainer_ext.py
View file @ fb5665b5
...@@ -28,6 +28,7 @@ from transformers.testing_utils import ( ...@@ -28,6 +28,7 @@ from transformers.testing_utils import (
execute_subprocess_async, execute_subprocess_async,
get_gpu_count, get_gpu_count,
get_torch_dist_unique_port, get_torch_dist_unique_port,
require_torch,
require_torch_gpu, require_torch_gpu,
require_torch_multi_gpu, require_torch_multi_gpu,
require_torch_non_multi_gpu, require_torch_non_multi_gpu,
...@@ -69,6 +70,7 @@ def require_apex(test_case): ...@@ -69,6 +70,7 @@ def require_apex(test_case):
return test_case return test_case
@require_torch
class TestTrainerExt(TestCasePlus): class TestTrainerExt(TestCasePlus):
def run_seq2seq_quick( def run_seq2seq_quick(
self, self,
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment