ExternalVendorCode/lollms
1
0
Fork 0
mirror of https://github.com/ParisNeo/lollms.git synced 2025-02-21 01:42:08 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

fixed models loading

Browse Source
This commit is contained in:
Saifeddine ALOUI 2024-02-29 17:44:00 +01:00
parent 989794aab0
commit 8d48f95264
1 changed files with 13 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
lollms/binding.py
View File

@ -124,8 +124,19 @@ class LLMBinding:
if "ggml" in str(mp).lower() or "gguf" in str(mp).lower():
if mp.is_dir():
for f in mp.iterdir():
if not "mmproj" in f.stem:
if not "mmproj" in f.stem and not f.is_dir():
return f
else:
show_message_dialog("Warning","I detected that your model was installed with previous format.\nI'll just migrate it to thre new format.\nThe new format allows you to have multiple model variants and also have the possibility to use multimodal models.")
model_root:Path = model_path.parent/model_path.stem
model_root.mkdir(exist_ok=True, parents=True)
shutil.move(model_path, model_root)
model_path = model_root/model_path.name
self.config.model_name = model_root.name
root_path = model_root
self.config.save_config()
return model_path
else:
return mp
@ -619,7 +630,7 @@ class LLMBinding:
ASCIIColors.yellow(model_path)
else:
return None
else:
else:
model_path = self.searchModelPath(self.config.model_name)
return model_path