Update app.py

app.py

@@ -66,12 +66,10 @@ device = 'cuda'
 dtype = torch.bfloat16
 
 try:
-   vae = AutoencoderKLWan.from_pretrained(MODEL_ID, subfolder="vae", torch_dtype=torch.float16)
+   vae = AutoencoderKLWan.from_pretrained(MODEL_ID, subfolder="vae", torch_dtype=torch.bfloat16)
    wan_path = hf_hub_download(repo_id=SUB_MODEL_ID, filename=SUB_MODEL_FILENAME)
    transformer = NagWanTransformer3DModel.from_single_file(wan_path, torch_dtype=torch.bfloat16)
-   pipe = NAGWanPipeline.from_pretrained(
-       MODEL_ID, vae=vae, transformer=transformer, torch_dtype=torch.bfloat16
-   )
+   pipe = NAGWanPipeline.from_pretrained(MODEL_ID, vae=vae, transformer=transformer, torch_dtype=torch.bfloat16)
    pipe.scheduler = UniPCMultistepScheduler.from_config(pipe.scheduler.config, flow_shift=5.0)
    pipe.to("cuda")
 
@@ -152,7 +150,7 @@ with gr.Blocks(theme=theme) as demo:
         with gr.Column():
             prompt = gr.Textbox(label="Prompt",lines=2,elem_classes="prompt-input")
             duration_seconds_input = gr.Slider(minimum=0.5,maximum=15,step=0.1,value=DEFAULT_DURATION_SECONDS,label="Duration",elem_classes="slider-container")
-            GPU_time_input = gr.Slider(value=181.0,minimum=0.0,maximum=300.0,step=1.0,label="GPU time (s)")
+            GPU_time_input = gr.Slider(value=171.0,minimum=0.0,maximum=300.0,step=1.0,label="GPU time (s)")
             randomize_seed_checkbox = gr.Checkbox(label="Random Seed",value=True,interactive=True)
             seed_input = gr.Slider(label="Seed",minimum=0,maximum=MAX_SEED,step=1,value=DEFAULT_SEED,interactive=True)