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

model_type_arch

Browse source
This commit is contained in:
Daniel Han 2025-03-19 03:03:43 -07:00
commit d2c3a87060
2 changed files with 4 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

1
unsloth/models/_utils.py
View file

@ -1177,6 +1177,7 @@ def unsloth_compile_transformers(
return
if disable: return
model_types = list(dict().fromkeys(model_types).keys())
for model_type in model_types:
_unsloth_compile_transformers(
model_type,

6
unsloth/models/vision.py
View file

@ -253,10 +253,10 @@ class FastBaseModel:
try: vllm_version = f" vLLM: {importlib_version('vllm')}."
except: vllm_version = ""
print(model_types)
model_type_arch = model_types[0]
if model_type_arch == "siglip" and len(model_types) != 1:
model_type_arch = model_types[1]
if model_type_arch == "siglip":
for model_type_arch in model_types:
if model_type_arch != "siglip": break
statistics = \
f"==((====))== Unsloth {__version__}: Fast {model_type_arch.title()} patching. Transformers: {transformers_version}.{vllm_version}\n"\