alphard/ComfyUI-Lora-Manager
1
0
Fork 0
mirror of https://github.com/willmiao/ComfyUI-Lora-Manager.git synced 2026-03-21 21:22:11 -03:00
Code Issues Packages Projects Releases Wiki Activity

fix(routes): await trained words extraction

Browse Source
This commit is contained in:
pixelpaws
2025-10-03 22:45:25 +08:00
parent 40e3c6134c
commit 2d37a7341a
1 changed files with 8 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
py/routes/handlers/misc_handlers.py
View File

@@ -320,9 +320,14 @@ class TrainedWordsHandler:
if not file_path.endswith(".safetensors"):
return web.json_response({"success": False, "error": "File must be a safetensors file"}, status=400)
trained_words = extract_trained_words(file_path)
sorted_words = sorted(trained_words, key=lambda w: w.get("count", 0), reverse=True)
return web.json_response({"success": True, "trained_words": sorted_words})
trained_words, class_tokens = await extract_trained_words(file_path)
return web.json_response(
{
"success": True,
"trained_words": trained_words,
"class_tokens": class_tokens,
}
)
except Exception as exc: # pragma: no cover - defensive logging
logger.error("Failed to get trained words: %s", exc, exc_info=True)
return web.json_response({"success": False, "error": str(exc)}, status=500)