From 769da2586efb20b3985ce0af0a38dbbc89ae70ce Mon Sep 17 00:00:00 2001 From: Ethanfel Date: Thu, 12 Feb 2026 23:45:13 +0100 Subject: [PATCH] =?UTF-8?q?Fix=20SGM-VFI=20auto-download:=20correct=20file?= =?UTF-8?q?=20extension=20.pth=20=E2=86=92=20.pkl?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit 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 --- README.md | 2 +- nodes.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/README.md b/README.md index 82b7051..4ad27aa 100644 --- a/README.md +++ b/README.md @@ -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 diff --git a/nodes.py b/nodes.py index 8c5aed7..5308435 100644 --- a/nodes.py +++ b/nodes.py @@ -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):