Update

README.md

@@ -7,6 +7,7 @@ sdk: gradio
 sdk_version: 5.44.1
 app_file: app.py
 pinned: false
+python_version: 3.10
 ---
 
 <h1 align="center">Transition Models: Rethinking the Generative Learning Objective</h1>

app.py

@@ -47,9 +47,7 @@ def load_model_components(device: str = "cuda"):
         from huggingface_hub import hf_hub_download
 
         ckpt_path = hf_hub_download(
-            repo_id="blanchon/TiM-checkpoints",
-            filename="t2i_model.bin",
-            local_dir=Path(__file__).parent,
+            repo_id="blanchon/TiM-checkpoints", filename="t2i_model.bin"
         )
 
         if not Path(config_path).exists():

requirements.txt

@@ -1,6 +1,6 @@
 gradio>=4.0.0
 spaces>=0.28.0
-torch>=2.1.0
+torch==2.8.0
 torchvision
 diffusers
 transformers>=4.25.0
@@ -10,6 +10,5 @@ numpy
 Pillow
 safetensors
 tqdm
-# flash-attn>=2.0.0
 flash-attn @ https://github.com/mjun0812/flash-attention-prebuild-wheels/releases/download/v0.3.14/flash_attn-2.8.2+cu129torch2.8-cp310-cp310-linux_x86_64.whl
 accelerate