mirror of
https://github.com/hiyouga/LLaMA-Factory.git
synced 2026-02-26 07:45:59 +08:00
[ci] fix ut huggingface hub 429 error when transformers>=5.0.0 (#10155)
This commit is contained in:
@@ -166,3 +166,33 @@ def _manage_distributed_env(request: FixtureRequest, monkeypatch: MonkeyPatch) -
|
|||||||
def fix_valuehead_cpu_loading():
|
def fix_valuehead_cpu_loading():
|
||||||
"""Fix valuehead model loading."""
|
"""Fix valuehead model loading."""
|
||||||
patch_valuehead_model()
|
patch_valuehead_model()
|
||||||
|
|
||||||
|
|
||||||
|
@pytest.fixture(scope="session", autouse=True)
|
||||||
|
def bypass_mistral_regex_check():
|
||||||
|
"""Disable Mistral regex network check.
|
||||||
|
|
||||||
|
Monkey-patch TokenizersBackend._patch_mistral_regex into a no-op.
|
||||||
|
"""
|
||||||
|
try:
|
||||||
|
from transformers.tokenization_utils_fast import TokenizersBackend
|
||||||
|
except ImportError:
|
||||||
|
# Very old transformers, nothing to patch
|
||||||
|
yield
|
||||||
|
return
|
||||||
|
|
||||||
|
if not hasattr(TokenizersBackend, "_patch_mistral_regex"):
|
||||||
|
# Method does not exist in this version
|
||||||
|
yield
|
||||||
|
return
|
||||||
|
|
||||||
|
# Backup original method
|
||||||
|
original = TokenizersBackend._patch_mistral_regex
|
||||||
|
|
||||||
|
# Replace with no-op
|
||||||
|
TokenizersBackend._patch_mistral_regex = lambda cls, tokenizer, *args, **kwargs: tokenizer
|
||||||
|
|
||||||
|
yield
|
||||||
|
|
||||||
|
# Restore original method
|
||||||
|
TokenizersBackend._patch_mistral_regex = original
|
||||||
|
|||||||
@@ -172,3 +172,33 @@ def _manage_distributed_env(request: FixtureRequest, monkeypatch: MonkeyPatch) -
|
|||||||
monkeypatch.setattr(torch.cuda, "device_count", lambda: 1)
|
monkeypatch.setattr(torch.cuda, "device_count", lambda: 1)
|
||||||
elif CURRENT_DEVICE == "npu":
|
elif CURRENT_DEVICE == "npu":
|
||||||
monkeypatch.setattr(torch.npu, "device_count", lambda: 1)
|
monkeypatch.setattr(torch.npu, "device_count", lambda: 1)
|
||||||
|
|
||||||
|
|
||||||
|
@pytest.fixture(scope="session", autouse=True)
|
||||||
|
def bypass_mistral_regex_check():
|
||||||
|
"""Disable Mistral regex network check.
|
||||||
|
|
||||||
|
Monkey-patch TokenizersBackend._patch_mistral_regex into a no-op.
|
||||||
|
"""
|
||||||
|
try:
|
||||||
|
from transformers.tokenization_utils_fast import TokenizersBackend
|
||||||
|
except ImportError:
|
||||||
|
# Very old transformers, nothing to patch
|
||||||
|
yield
|
||||||
|
return
|
||||||
|
|
||||||
|
if not hasattr(TokenizersBackend, "_patch_mistral_regex"):
|
||||||
|
# Method does not exist in this version
|
||||||
|
yield
|
||||||
|
return
|
||||||
|
|
||||||
|
# Backup original method
|
||||||
|
original = TokenizersBackend._patch_mistral_regex
|
||||||
|
|
||||||
|
# Replace with no-op
|
||||||
|
TokenizersBackend._patch_mistral_regex = lambda cls, tokenizer, *args, **kwargs: tokenizer
|
||||||
|
|
||||||
|
yield
|
||||||
|
|
||||||
|
# Restore original method
|
||||||
|
TokenizersBackend._patch_mistral_regex = original
|
||||||
|
|||||||
Reference in New Issue
Block a user