remove 512

app.py

@@ -64,27 +64,27 @@ def infer(prompt, output_size, seed, num_frames, ddim_steps, unconditional_guida
         model = model_1024
         fps = 28
         num_frames = min(num_frames, 36)
-    elif output_size == "256x256":
-        width = 256 
-        height = 256
-        ckpt_dir_256 = "checkpoints/base_256_v1"
-        ckpt_path_256 = "checkpoints/base_256_v1/model_256.pth"
-        config_256 = "configs/inference_t2v_tconv256_v1.0_freenoise.yaml"
-        config_256 = OmegaConf.load(config_256)
-        model_config_256 = config_256.pop("model", OmegaConf.create())
-        model_256 = instantiate_from_config(model_config_256)
-        model_256 = model_256.cuda()
-        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)
-        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
+    # elif output_size == "256x256":
+    #     width = 256 
+    #     height = 256
+    #     ckpt_dir_256 = "checkpoints/base_256_v1"
+    #     ckpt_path_256 = "checkpoints/base_256_v1/model_256.pth"
+    #     config_256 = "configs/inference_t2v_tconv256_v1.0_freenoise.yaml"
+    #     config_256 = OmegaConf.load(config_256)
+    #     model_config_256 = config_256.pop("model", OmegaConf.create())
+    #     model_256 = instantiate_from_config(model_config_256)
+    #     model_256 = model_256.cuda()
+    #     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)
+    #     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
 
     if seed is None:
         seed = int.from_bytes(os.urandom(2), "big")
@@ -287,11 +287,12 @@ with gr.Blocks(css=css) as demo:
             with gr.Accordion('FreeNoise Parameters (feel free to adjust these parameters based on your prompt): ', open=False):
                 with gr.Row():
                     # output_size = gr.Dropdown(["320x512", "576x1024"], value="320x512", label="Output Size", info="250s for 512 model, 900s for 1024 model (32 frames). Recovering from sleeping will take more time to download ckpt")
-                    output_size = gr.Dropdown(["256x256", "576x1024"], value="576x1024", label="Output Size", info="900s for 1024 model (32 frames). Recovering from sleeping will take more time to download ckpt")
+                    # output_size = gr.Dropdown(["256x256", "576x1024"], value="576x1024", label="Output Size", info="900s for 1024 model (32 frames). Recovering from sleeping will take more time to download ckpt")
+                    output_size = gr.Dropdown(["576x1024"], value="576x1024", label="Output Size", info="900s for 1024 model (32 frames). Recovering from sleeping will take more time to download ckpt")
                 with gr.Row():
                     num_frames = gr.Slider(label='Frames (a multiple of 4), max 36 for 1024 model',
                              minimum=16,
-                             maximum=64,
+                             maximum=36,
                              step=4,
                              value=32)
                     ddim_steps = gr.Slider(label='DDIM Steps',