Update __init__.py

This commit is contained in:
Saifeddine ALOUI 2023-11-15 15:06:06 +01:00 committed by GitHub
parent 14327e7541
commit 550c143bfd
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -1,6 +1,6 @@
###### ######
# Project : lollms-webui # Project : lollms-webui
# File : api.py # File : api/__init__.py
# Author : ParisNeo with the help of the community # Author : ParisNeo with the help of the community
# license : Apache 2.0 # license : Apache 2.0
# Description : # Description :
@ -430,7 +430,7 @@ class LoLLMsAPPI(LollmsApplication):
installation_dir = self.binding.searchModelParentFolder(model_path) installation_dir = self.binding.searchModelParentFolder(model_path)
binding_folder = self.config["binding_name"] binding_folder = self.config["binding_name"]
if model_type=="gptq": if model_type=="gptq" or model_type=="awq":
filename = model_path.split("/")[4] filename = model_path.split("/")[4]
installation_path = installation_dir / filename installation_path = installation_dir / filename
else: else: