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

Update __init__.py

Browse Source
This commit is contained in:
Saifeddine ALOUI 2023-11-10 17:03:57 +01:00 committed by GitHub
parent 614bbb3b25
commit 16d712abff
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
api/__init__.py
View File

@ -256,7 +256,7 @@ class LoLLMsAPPI(LollmsApplication):
model_type:str=data["type"]
progress = 0
installation_dir = self.binding.searchModelParentFolder(model_path.split('/')[-1], model_type)
if model_type=="gptq":
if model_type=="gptq" or model_type=="awq":
parts = model_path.split("/")
if len(parts)==2:
filename = parts[1]
@ -298,6 +298,7 @@ class LoLLMsAPPI(LollmsApplication):
'speed': self.download_infos[signature]['speed'],
}, room=room_id
)
return
socketio.emit('install_progress',{
'status': True,