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

Fix SGM-VFI auto-download: correct file extension .pth → .pkl

Browse Source 
The Google Drive folder contains .pkl files but the default model
name used .pth, causing the post-download existence check to fail.

Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
This commit is contained in:
Ethanfel
2026-02-12 23:45:13 +01:00
parent d935462e24
commit 769da2586e
2 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
README.md
View File

@@ -96,7 +96,7 @@ Loads an SGM-VFI checkpoint. Auto-downloads from Google Drive on first use to `C
Available checkpoints:
| Checkpoint | Variant | Params |
|-----------|---------|--------|
| `ours-1-2-points.pth` | Small | ~15M + GMFlow |
| `ours-1-2-points.pkl` | Small | ~15M + GMFlow |
#### SGM-VFI Interpolate

2
nodes.py
View File

@@ -34,7 +34,7 @@ if not os.path.exists(EMA_MODEL_DIR):
# Google Drive folder ID for SGM-VFI pretrained models
SGM_GDRIVE_FOLDER_ID = "1S5O6W0a7XQDHgBtP9HnmoxYEzWBIzSJq"
SGM_DEFAULT_MODEL = "ours-1-2-points.pth"
SGM_DEFAULT_MODEL = "ours-1-2-points.pkl"
SGM_MODEL_DIR = os.path.join(folder_paths.models_dir, "sgm-vfi")
if not os.path.exists(SGM_MODEL_DIR):