riaz.somc/transformers
1
0
Fork 0
mirror of https://github.com/huggingface/transformers.git synced 2025-08-03 03:31:05 +06:00
Code Issues Actions 2 Packages Projects Releases Wiki Activity

Fix hub_retry (#36449)

Browse Source 
* cry

* trigger

---------

Co-authored-by: ydshieh <ydshieh@users.noreply.github.com>
This commit is contained in:
Yih-Dar 2025-02-27 14:38:25 +01:00 committed by GitHub
parent 6a876462c3
commit 482d17be60
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
tests/test_modeling_common.py
View File

@ -223,7 +223,7 @@ class ModelTesterMixin:
if attr_name.startswith("test_"): if attr_name.startswith("test_"):
attr = getattr(cls, attr_name) attr = getattr(cls, attr_name)
if callable(attr): if callable(attr):
setattr(cls, attr_name, hub_retry(attr)) setattr(cls, attr_name, hub_retry()(attr))
@property @property
def all_generative_model_classes(self): def all_generative_model_classes(self):