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

Update vision.py

Browse source
This commit is contained in:
Daniel Han 2025-05-15 02:07:29 -07:00
commit 633221f8c9
1 changed files with 2 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

6
unsloth/models/vision.py
View file

@ -320,7 +320,6 @@ class FastBaseModel:
dtype = torch.float32
bnb_compute_dtype = eval(bnb_compute_dtype)
pass
print("bnb_compute_dtype", bnb_compute_dtype)
# Stop SDPA for some archs like Pixtral / Mistral3
if not ("attn_implementation" in kwargs):
@ -388,9 +387,8 @@ class FastBaseModel:
# Edit data-types
if custom_datatype is not None:
with torch.inference_mode():
for name, module in model.named_modules():
exec(custom_datatype)
for name, module in model.named_modules():
exec(custom_datatype)
pass
# Counteract saved tokenizers