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

Update loader.py

Browse source
This commit is contained in:
Daniel Han 2024-11-26 00:35:05 -08:00
commit a02de20cb8
1 changed files with 2 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

2
unsloth/models/loader.py
View file

@ -406,6 +406,7 @@ class FastVisionModel(FastBaseVisionModel):
is_model = True
except Exception as error:
autoconfig_error = str(error)
print(autoconfig_error)
is_model = False
try:
peft_config = PeftConfig.from_pretrained(
@ -418,6 +419,7 @@ class FastVisionModel(FastBaseVisionModel):
is_peft = True
except Exception as error:
peft_error = str(error)
print(peft_error)
is_peft = False
pass