Ethanfel/ComfyUI-Tween
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Reuse FlashVSR models from 1038lab node if already downloaded

Browse Source 
Check models/FlashVSR/ (1038lab convention) before models/flashvsr/ to
avoid downloading ~7GB of checkpoints twice. Only create the directory
when actually downloading.

Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
This commit is contained in:
Ethanfel
2026-02-13 15:42:10 +01:00
parent 0fecfcee37
commit 8317a0603e
1 changed files with 13 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
nodes.py
View File

@@ -1522,9 +1522,18 @@ FLASHVSR_REQUIRED_FILES = [
"Prompt.safetensors", "Prompt.safetensors",
] ]
FLASHVSR_MODEL_DIR = os.path.join(folder_paths.models_dir, "flashvsr") # Check common locations so we reuse models from 1038lab/ComfyUI-FlashVSR
if not os.path.exists(FLASHVSR_MODEL_DIR): FLASHVSR_MODEL_DIR = None
os.makedirs(FLASHVSR_MODEL_DIR, exist_ok=True) for _dirname in ("FlashVSR", "flashvsr"):
_candidate = os.path.join(folder_paths.models_dir, _dirname)
if os.path.isdir(_candidate) and all(
os.path.exists(os.path.join(_candidate, f)) for f in FLASHVSR_REQUIRED_FILES
):
FLASHVSR_MODEL_DIR = _candidate
break
if FLASHVSR_MODEL_DIR is None:
# Default to "FlashVSR" (matches 1038lab convention)
FLASHVSR_MODEL_DIR = os.path.join(folder_paths.models_dir, "FlashVSR")
def download_flashvsr_models(model_dir): def download_flashvsr_models(model_dir):
@@ -1536,6 +1545,7 @@ def download_flashvsr_models(model_dir):
if not missing: if not missing:
return return
os.makedirs(model_dir, exist_ok=True)
logger.info(f"[FlashVSR] Missing files: {', '.join(missing)}. Downloading from HuggingFace...") logger.info(f"[FlashVSR] Missing files: {', '.join(missing)}. Downloading from HuggingFace...")
snapshot_download( snapshot_download(
repo_id=FLASHVSR_HF_REPO, repo_id=FLASHVSR_HF_REPO,