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):