Update __init__.py

ParisNeo · Nov 15, 2023 · 550c143 · 550c143
1 parent 14327e7
commit 550c143
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/api/__init__.py b/api/__init__.py
@@ -1,6 +1,6 @@
 ######
 # Project       : lollms-webui
-# File          : api.py
+# File          : api/__init__.py
 # Author        : ParisNeo with the help of the community
 # license       : Apache 2.0
 # Description   : 
@@ -430,7 +430,7 @@ def uninstall_model(data):
             installation_dir = self.binding.searchModelParentFolder(model_path)
 
             binding_folder = self.config["binding_name"]
-            if model_type=="gptq":
+            if model_type=="gptq" or  model_type=="awq":
                 filename = model_path.split("/")[4]
                 installation_path = installation_dir / filename
             else: