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

Enhance trained words extraction and display: include class tokens in response and update UI accordingly. See #147

Browse Source
This commit is contained in:
Will Miao
2025-06-04 12:03:36 +08:00
parent 4b96c650eb
commit b4e7feed06
4 changed files with 167 additions and 46 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
py/routes/misc_routes.py
View File

@@ -1005,13 +1005,14 @@ class MiscRoutes:
'error': 'File is not a safetensors file'
}, status=400)
# Extract trained words
trained_words = await extract_trained_words(file_path)
# Extract trained words and class_tokens
trained_words, class_tokens = await extract_trained_words(file_path)
# Return result
# Return result with both trained words and class tokens
return web.json_response({
'success': True,
'trained_words': trained_words
'trained_words': trained_words,
'class_tokens': class_tokens
})
except Exception as e: