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

Refactor progress reporting in DownloadManager and improve selection logic in MoveManager

Browse Source
This commit is contained in:
Will Miao
2025-03-03 18:51:35 +08:00
parent 81b401106f
commit a38757fee6
2 changed files with 16 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
py/services/download_manager.py
View File

@@ -80,7 +80,7 @@ class DownloadManager:
if images:
# Report preview download progress
if progress_callback:
await progress_callback(5) # 5% progress for starting preview download
await progress_callback(1) # 1% progress for starting preview download
preview_ext = '.mp4' if images[0].get('type') == 'video' else '.png'
preview_path = os.path.splitext(save_path)[0] + '.preview' + preview_ext
@@ -91,7 +91,7 @@ class DownloadManager:
# Report preview download completion
if progress_callback:
await progress_callback(10) # 10% progress after preview download
await progress_callback(3) # 3% progress after preview download
# Download model file with progress tracking
success, result = await self.civitai_client._download_file(
@@ -149,6 +149,6 @@ class DownloadManager:
progress_callback: Callback function for progress updates
"""
if progress_callback:
# Scale file progress to 10-100 range (after preview download)
overall_progress = 10 + (file_progress * 0.9) # 90% of progress for file download
# Scale file progress to 3-100 range (after preview download)
overall_progress = 3 + (file_progress * 0.97) # 97% of progress for file download
await progress_callback(round(overall_progress))