ExternalVendorCode/lollms
1
0
Fork 0
mirror of https://github.com/ParisNeo/lollms.git synced 2024-12-20 21:23:17 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

upgraded

Browse Source
This commit is contained in:
Saifeddine ALOUI 2023-10-22 22:17:54 +02:00
parent cc909a7dda
commit 19c8baada6
1 changed files with 19 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

23
lollms/main_config.py
View File

@ -142,17 +142,32 @@ class LOLLMSConfig(BaseConfig):
def searchModelPath(self, model_name:str): def searchModelPath(self, model_name:str):
model_path=None model_path=None
for mn in self.lollms_paths.binding_models_paths: for mn in self.models_folders:
if mn.name in model_name.lower(): if mn.name in model_name.lower():
model_path = mn/model_name if mn.name == "ggml":
try:
idx = model_name.index("-GGML")
models=[m for m in mn.iterdir() if model_name[:idx].lower() in m.name.lower()]
model_path = mn/models[0].name
except:
model_path = mn/model_name
elif mn.name == "gguf":
try:
idx = model_name.index("-GGUF")
models=[m for m in mn.iterdir() if model_name[:idx].lower() in m.name.lower()]
model_path = mn/models[0].name
except:
model_path = mn/model_name
else:
model_path = mn/model_name
break break
if model_path is None: if model_path is None:
model_path = self.lollms_paths.binding_models_paths[0]/model_name model_path = self.models_folders[0]/model_name
return model_path return model_path
def download_model(self, url, binding, callback = None): def download_model(self, url, binding, callback = None):
model_name = url.split("/")[-1] model_name = url.split("/")[-1]
folder_path = self.searchModelPath(model_name) folder_path = binding.searchModelPath(model_name)
model_full_path = (folder_path / model_name) model_full_path = (folder_path / model_name)
if binding is not None and hasattr(binding,'download_model'): if binding is not None and hasattr(binding,'download_model'):
binding.download_model(url, model_full_path, callback) binding.download_model(url, model_full_path, callback)