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

fixed disk usage problem

Browse Source
This commit is contained in:
Saifeddine ALOUI 2023-06-14 13:12:27 +02:00
parent 888a5f81c0
commit 23b60fdf25
1 changed files with 18 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

29
app.py
View File

@ -505,21 +505,28 @@ class LoLLMsWebUI(LoLLMsAPPI):
current_drive = Path.cwd().anchor current_drive = Path.cwd().anchor
drive_disk_usage = psutil.disk_usage(current_drive) drive_disk_usage = psutil.disk_usage(current_drive)
try: try:
models_folder_disk_usage = psutil.disk_usage(self.lollms_paths.personal_models_path/f'{self.config["binding_name"]}') models_folder_disk_usage = psutil.disk_usage(str(self.lollms_paths.personal_models_path/f'{self.config["binding_name"]}'))
return jsonify({ return jsonify({
"total_space":drive_disk_usage.total, "total_space":drive_disk_usage.total,
"available_space":drive_disk_usage.free, "available_space":drive_disk_usage.free,
"usage":drive_disk_usage.used,
"percent_usage":models_folder_disk_usage.used,
"binding_models_usage": models_folder_disk_usage.used
})
except:
return jsonify({
"total_space":drive_disk_usage.total,
"available_space":drive_disk_usage.free,
"percent_usage":drive_disk_usage.percent, "percent_usage":drive_disk_usage.percent,
"models_folder_usage": None
"binding_disk_total_space":models_folder_disk_usage.total,
"binding_disk_available_space":drive_disk_usage.free,
"binding_models_usage": models_folder_disk_usage.used,
"binding_models_percent_usage": models_folder_disk_usage.percent,
})
except Exception as ex:
return jsonify({
"total_space":drive_disk_usage.total,
"available_space":drive_disk_usage.free,
"percent_usage":drive_disk_usage.percent,
"binding_disk_total_space": None,
"binding_disk_available_space": None,
"binding_models_usage": None,
"binding_models_percent_usage": None,
}) })
def list_bindings(self): def list_bindings(self):