ceos/unsloth
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Handle transformers rename from PretrainedConfig to PreTrainedConfig (#3445)

Browse source
This commit is contained in:
DoubleMathew 2025-10-13 18:00:28 -05:00 • committed by GitHub
commit 2a46dd212d
2 changed files with 13 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

7
unsloth/models/_utils.py
View file

@ -432,7 +432,12 @@ try:
except:
pass
from transformers import __version__ as transformers_version
from transformers import PretrainedConfig
try:
from transformers import PreTrainedConfig
except:
from transformers import PretrainedConfig
model_architectures = ["llama", "mistral", "gemma", "gemma2", "qwen2", "granite", "qwen3", "qwen3_moe", "falcon_h1"]
for model_name in model_architectures:

8
unsloth/models/vision.py
View file

@ -92,7 +92,13 @@ PRE_COMPILE_INFERENCE = [
"gpt_oss",
]
from transformers import GenerationConfig, CompileConfig, HybridCache, AutoConfig, PretrainedConfig
from transformers import GenerationConfig, CompileConfig, HybridCache, AutoConfig
try:
from transformers import PreTrainedConfig
PretrainedConfig = PreTrainedConfig
except:
from transformers import PretrainedConfig
HAS_TORCH_DTYPE = "torch_dtype" in PretrainedConfig.__doc__
from transformers import GenerationConfig, CompileConfig, HybridCache