Update hf checkpoint links

README.md

@@ -10,7 +10,7 @@
     <a href="https://linkedin.com/in/dmitriyvatolin">Dmitriy Vatolin</a>
   </p>
   <h3 align="center">ICCV 2025</h3>
-  <h3 align="center"><a href="https://arxiv.org/abs/2506.23151">📄 Paper</a> | <a href="https://msu-video-group.github.io/memfof">🌐 Project Page</a> | <a href="https://colab.research.google.com/github/msu-video-group/memfof/blob/dev/demo.ipynb">🚀 Colab</a> | <a href="https://huggingface.co/spaces/egorchistov/MEMFOF">🤗 Demo</a></h3>
+  <h3 align="center"><a href="https://arxiv.org/abs/2506.23151">📄 Paper</a> | <a href="https://msu-video-group.github.io/memfof">🌐 Project Page</a> | <a href="https://colab.research.google.com/github/msu-video-group/memfof/blob/dev/demo.ipynb">🚀 Colab</a> | <a href="https://huggingface.co/spaces/egorchistov/optical-flow-MEMFOF">🤗 Demo</a> | <a href="https://huggingface.co/collections/egorchistov/optical-flow-memfof-685695802e71b207b96d8fb8">📦 Models</a></h3>
 </p>
 
 ## 🏅 Overview
@@ -48,12 +48,12 @@ python3 demo.py
 
 ## 📦 Models
 
-- [`MEMFOF-Tartan`](https://huggingface.co/egorchistov/MEMFOF-Tartan)
-- [`MEMFOF-Tartan-T`](https://huggingface.co/egorchistov/MEMFOF-Tartan-T)
-- [`MEMFOF-Tartan-T-TSKH`](https://huggingface.co/egorchistov/MEMFOF-Tartan-T-TSKH) (✅ Recommended for real-world videos)
-- [`MEMFOF-Tartan-T-TSKH-kitti`](https://huggingface.co/egorchistov/MEMFOF-Tartan-T-TSKH-kitti)
-- [`MEMFOF-Tartan-T-TSKH-sintel`](https://huggingface.co/egorchistov/MEMFOF-Tartan-T-TSKH-sintel)
-- [`MEMFOF-Tartan-T-TSKH-spring`](https://huggingface.co/egorchistov/MEMFOF-Tartan-T-TSKH-spring)
+- [`MEMFOF-Tartan`](https://huggingface.co/egorchistov/optical-flow-MEMFOF-Tartan)
+- [`MEMFOF-Tartan-T`](https://huggingface.co/egorchistov/optical-flow-MEMFOF-Tartan-T)
+- [`MEMFOF-Tartan-T-TSKH`](https://huggingface.co/egorchistov/optical-flow-MEMFOF-Tartan-T-TSKH) (✅ Best for real videos)
+- [`MEMFOF-Tartan-T-TSKH-kitti`](https://huggingface.co/egorchistov/optical-flow-MEMFOF-Tartan-T-TSKH-kitti)
+- [`MEMFOF-Tartan-T-TSKH-sintel`](https://huggingface.co/egorchistov/optical-flow-MEMFOF-Tartan-T-TSKH-sintel)
+- [`MEMFOF-Tartan-T-TSKH-spring`](https://huggingface.co/egorchistov/optical-flow-MEMFOF-Tartan-T-TSKH-spring)
 
 ## 🗂️ Datasets
 

demo.ipynb

@@ -38,7 +38,7 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "model = MEMFOF.from_pretrained(\"egorchistov/MEMFOF-Tartan-T-TSKH\").eval().to(device)"
+    "model = MEMFOF.from_pretrained(\"egorchistov/optical-flow-MEMFOF-Tartan-T-TSKH\").eval().to(device)"
    ]
   },
   {

demo.py

@@ -151,7 +151,7 @@ def download(url: str) -> str:
 @spaces.GPU(duration=60)
 def run_demo(input_path: str, model_name: str) -> str:
     device = torch.device("cuda") if torch.cuda.is_available() else torch.device("cpu")
-    model = MEMFOF.from_pretrained(f"egorchistov/{model_name}").eval().to(device)
+    model = MEMFOF.from_pretrained(f"egorchistov/optical-flow-{model_name}").eval().to(device)
     output_path = tempfile.NamedTemporaryFile(suffix=".mp4").name
     process_video(model, input_path, output_path, device, progress=gr.Progress(), soft_duration=57)
     return output_path

scripts/ckpts.py

@@ -26,7 +26,7 @@ def main():
         if os.path.exists(output_path):
             continue
         print(f"Downloading {output_path}")
-        snapshot_path = snapshot_download(repo_id=f"egorchistov/MEMFOF-{ckpt}")
+        snapshot_path = snapshot_download(repo_id=f"egorchistov/optical-flow-MEMFOF-{ckpt}")
         state_dict = load_file(os.path.join(snapshot_path, "model.safetensors"))
         state_dict = {"model." + k: v for k, v in state_dict.items()}
         torch.save({"state_dict": state_dict}, output_path)