Fix some tests #35682
Fix some tests #35682
Conversation
Cyrilvallez
requested review from
ArthurZucker and
Rocketknight1
as code owners
|
The docs for this PR live here. All of your documentation changes will be reflected on that endpoint. The docs are available until 30 days after the last update. |
src/transformers/testing_utils.py
Outdated
| def require_torch_large_gpu(test_case, memory: float = 20): | ||
| """Decorator marking a test that requires a GPU with more than `memory` GiB of memory.""" | ||
| if torch_device != "cuda": | ||
| return unittest.skip(reason=f"test requires a GPU with more than {memory} GiB of memory")(test_case) | ||
|
|
||
| return unittest.skipUnless( | ||
| torch.cuda.get_device_properties(0).total_memory / 1024**3 > memory, | ||
| f"test requires a GPU with more than {memory} GiB of memory", | ||
| )(test_case) |
There was a problem hiding this comment.
we can also just use the @tooslow 😉
|
From what I see, |
|
Yes, CI will run A10 on some special events. Let's go with this new |
|
I have to double check if the CI bot has the access to those repo. decorated with Update: |
|
Thanks for double-checking @ydshieh! I applied your comments. Merging then as we agree! |
* cohere tests * glm tests * cohere2 model name * create decorator * update * fix cohere2 completions * style * style * style * add cuda in comments
* cohere tests * glm tests * cohere2 model name * create decorator * update * fix cohere2 completions * style * style * style * add cuda in comments
What does this PR do?
Fixes some tests: