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

fix: enhance model processing logic to skip already processed models only if their directories contain files. See #259

Browse Source
This commit is contained in:
Will Miao
2025-06-27 13:09:19 +08:00
parent 5de1c8aa82
commit b1e2e3831f
1 changed files with 8 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
py/utils/example_images_download_manager.py
View File

@@ -295,10 +295,15 @@ class DownloadManager:
# Update current model info # Update current model info
download_progress['current_model'] = f"{model_name} ({model_hash[:8]})" download_progress['current_model'] = f"{model_name} ({model_hash[:8]})"
# Skip if already processed # Skip if already processed AND directory exists with files
if model_hash in download_progress['processed_models']: if model_hash in download_progress['processed_models']:
logger.debug(f"Skipping already processed model: {model_name}") model_dir = os.path.join(output_dir, model_hash)
return False has_files = os.path.exists(model_dir) and any(os.listdir(model_dir))
if has_files:
logger.debug(f"Skipping already processed model: {model_name}")
return False
else:
logger.info(f"Model {model_name} marked as processed but folder empty or missing, reprocessing")
# Create model directory # Create model directory
model_dir = os.path.join(output_dir, model_hash) model_dir = os.path.join(output_dir, model_hash)