add 512

app.py

@@ -34,7 +34,11 @@ def infer(prompt, output_size, seed, num_frames, ddim_steps, unconditional_guida
         if not os.path.exists(ckpt_path_512):
             os.makedirs(ckpt_dir_512, exist_ok=True)
             hf_hub_download(repo_id="MoonQiu/LongerCrafter", filename="model_512.ckpt", local_dir=ckpt_dir_512)
-        model_512 = load_model_checkpoint(model_512, ckpt_path_512)
+        try:
+            model_512 = load_model_checkpoint(model_512, ckpt_path_512)
+        except:
+            hf_hub_download(repo_id="MoonQiu/LongerCrafter", filename="model_512.ckpt", local_dir=ckpt_dir_512, force_download=True)
+            model_512 = load_model_checkpoint(model_512, ckpt_path_512)
         model_512.eval()
         model = model_512
         fps = 8
@@ -51,7 +55,11 @@ def infer(prompt, output_size, seed, num_frames, ddim_steps, unconditional_guida
         if not os.path.exists(ckpt_path_1024):
             os.makedirs(ckpt_dir_1024, exist_ok=True)
             hf_hub_download(repo_id="VideoCrafter/Text2Video-1024", filename="model.ckpt", local_dir=ckpt_dir_1024)
-        model_1024 = load_model_checkpoint(model_1024, ckpt_path_1024)
+        try:
+            model_1024 = load_model_checkpoint(model_1024, ckpt_path_1024)
+        except:
+            hf_hub_download(repo_id="VideoCrafter/Text2Video-1024", filename="model.ckpt", local_dir=ckpt_dir_1024, force_download=True)
+            model_1024 = load_model_checkpoint(model_1024, ckpt_path_1024)
         model_1024.eval()
         model = model_1024
         fps = 28
@@ -69,7 +77,11 @@ def infer(prompt, output_size, seed, num_frames, ddim_steps, unconditional_guida
     #     if not os.path.exists(ckpt_path_256):
     #         os.makedirs(ckpt_dir_256, exist_ok=True)
     #         hf_hub_download(repo_id="MoonQiu/LongerCrafter", filename="model_256.pth", local_dir=ckpt_dir_256)
-    #     model_256 = load_model_checkpoint(model_256, ckpt_path_256)
+    #     try:
+    #         model_256 = load_model_checkpoint(model_256, ckpt_path_256)
+    #     except:
+    #         hf_hub_download(repo_id="MoonQiu/LongerCrafter", filename="model_256.pth", local_dir=ckpt_dir_256, force_download=True)
+    #         model_256 = load_model_checkpoint(model_256, ckpt_path_256)
     #     model_256.eval()
     #     model = model_256
     #     fps = 8
@@ -262,8 +274,10 @@ with gr.Blocks(css=css) as demo:
             Tuning-Free Longer Video Diffusion via Noise Rescheduling
             </p>
             <p style="text-align: center;">
-            [Arxiv](https://arxiv.org/abs/2310.15169) | [Project Page](http://haonanqiu.com/projects/FreeNoise.html) | [Github](https://github.com/arthur-qiu/LongerCrafter) <br />
-            </p>            
+            <a href="http://haonanqiu.com/projects/FreeNoise.html" target="_blank"><b>[Project Page]</b></a> &nbsp;&nbsp;&nbsp;&nbsp;
+            <a href="https://arxiv.org/abs/2310.15169" target="_blank"><b>[arXiv]</b></a> &nbsp;&nbsp;&nbsp;&nbsp;
+            <a href="https://github.com/arthur-qiu/LongerCrafter" target="_blank"><b>[Code]</b></a>
+            </p>         
             """
         )