TBNilles/ComfyUI-Model-Manager
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Skip hidden model files (#64)

Browse Source
This commit is contained in:
Hayden
2024-11-28 12:01:55 +08:00
committed by GitHub
parent 8b6c6ebdea
commit e8916307aa
5 changed files with 51 additions and 45 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
__init__.py
View File

@@ -130,7 +130,7 @@ async def read_models(request):
Scan all models and read their information.
"""
try:
result = services.scan_models()
result = services.scan_models(request)
return web.json_response({"success": True, "data": result})
except Exception as e:
error_msg = f"Read models failed: {str(e)}"
@@ -232,7 +232,7 @@ async def download_model_info(request):
post = await utils.get_request_body(request)
try:
scan_mode = post.get("scanMode", "diff")
await services.download_model_info(scan_mode)
await services.download_model_info(scan_mode, request)
return web.json_response({"success": True})
except Exception as e:
error_msg = f"Download model info failed: {str(e)}"
@@ -288,7 +288,7 @@ async def migrate_legacy_information(request):
Migrate legacy information.
"""
try:
await services.migrate_legacy_information()
await services.migrate_legacy_information(request)
return web.json_response({"success": True})
except Exception as e:
error_msg = f"Migrate model info failed: {str(e)}"