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

Add preview image scanning logic

Browse Source
This commit is contained in:
Will Miao
2025-01-29 07:21:33 +08:00
parent f076b97d21
commit a3235e4b3b
1 changed files with 3 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
nodes.py
View File

@@ -54,11 +54,10 @@ class LorasEndpoint:
loras = [] loras = []
for root, _, files in os.walk(self.loras_root): for root, _, files in os.walk(self.loras_root):
safetensors_files = [f for f in files if f.endswith('.safetensors')] safetensors_files = [f for f in files if f.endswith('.safetensors')]
# total_files = len(safetensors_files) total_files = len(safetensors_files)
for idx, filename in enumerate(safetensors_files, 1): for idx, filename in enumerate(safetensors_files, 1):
# self.send_progress(idx, total_files, f"Scanning: {filename}") self.send_progress(idx, total_files, f"Scanning: {filename}")
print(f"Scanning: {idx} {filename}")
file_path = os.path.join(root, filename) file_path = os.path.join(root, filename)
@@ -82,7 +81,7 @@ class LorasEndpoint:
loras.append(lora_data) loras.append(lora_data)
# self.send_progress(total_files, total_files, "Scan completed") self.send_progress(total_files, total_files, "Scan completed")
return loras return loras