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

fixed uninstall model

Browse Source
This commit is contained in:
Saifeddine ALOUI 2023-10-20 00:23:16 +02:00
parent 34f04bb427
commit a1dc373d15
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
api/__init__.py
View File

@ -428,7 +428,7 @@ class LoLLMsAPPI(LollmsApplication):
def uninstall_model(data):
model_path = data['path']
model_type:str=data.get("type","ggml")
installation_dir = self.lollms_paths.personal_models_path/self.config["binding_name"]
installation_dir = self.binding.searchModelParentFolder()
binding_folder = self.config["binding_name"]
if model_type=="gptq":
@ -450,7 +450,7 @@ class LoLLMsAPPI(LollmsApplication):
if not installation_path.exists():
# Try to find a version
model_path = installation_path.name.lower().replace("-ggml","").replace("-gguf","")
candidates = [m for m in (self.lollms_paths.personal_models_path/self.binding.binding_folder_name).iterdir() if model_path in m.name]
candidates = [m for m in installation_dir.iterdir() if model_path in m.name]
if len(candidates)>0:
model_path = candidates[0]
installation_path = model_path