Update app.py

app.py

@@ -15,7 +15,7 @@ is_canonical = True if os.environ.get("SPACE_ID") == "Pyramid-Flow/pyramid-flow"
 # Constants
 MODEL_PATH = "pyramid-flow-model"
 MODEL_REPO = "rain1011/pyramid-flow-sd3"
-MODEL_VARIANT = "diffusion_transformer_384p"
+MODEL_VARIANT = "diffusion_transformer_768p"
 MODEL_DTYPE = "bf16"
 
 def center_crop(image, target_width, target_height):
@@ -78,7 +78,7 @@ def generate_video(prompt, image=None, duration=10, guidance_scale=9, video_guid
                 guidance_scale=7.0,
                 video_guidance_scale=video_guidance_scale,
                 output_type="pil",
-                save_memory=True,
+                # save_memory=True,
             )
     else:
         with torch.no_grad(), torch.cuda.amp.autocast(enabled=True, dtype=torch_dtype):
@@ -86,13 +86,13 @@ def generate_video(prompt, image=None, duration=10, guidance_scale=9, video_guid
                 prompt=prompt,
                 num_inference_steps=[20, 20, 20],
                 video_num_inference_steps=[10, 10, 10],
-                height=320,
-                width=480,
+               height=768,     
+        width=1280,
                 temp=temp,
                 guidance_scale=guidance_scale,
                 video_guidance_scale=video_guidance_scale,
                 output_type="pil",
-                save_memory=True,
+                # save_memory=True,
             )
     output_path = f"{str(uuid.uuid4())}_output_video.mp4"
     export_to_video(frames, output_path, fps= 8)