Working version of wan2.1 diffuser - use this as benchmark

app.py

@@ -9,7 +9,6 @@ from diffusers.schedulers.scheduling_unipc_multistep import UniPCMultistepSchedu
 from diffusers.schedulers.scheduling_flow_match_euler_discrete import FlowMatchEulerDiscreteScheduler
 from huggingface_hub import hf_hub_download
 from lycoris import create_lycoris_from_weights
-from transformers import AutoConfig
 
 # Define model options
 MODEL_OPTIONS = {
@@ -70,12 +69,9 @@ def generate_video(
         seed = int(seed)
     
     torch.manual_seed(seed)
-
-    # Load config from the model
-    config = AutoConfig.from_pretrained(model_id)
     
-    vae = AutoencoderKLWan.from_pretrained(model_id, subfolder="vae", config=config, torch_dtype=torch.float32)
-    pipe = WanPipeline.from_pretrained(model_id, vae=vae, config=config, torch_dtype=torch.float16)
+    vae = AutoencoderKLWan.from_pretrained(model_id, subfolder="vae", torch_dtype=torch.float32)
+    pipe = WanPipeline.from_pretrained(model_id, vae=vae, torch_dtype=torch.float16)
     
     if scheduler_type == "UniPCMultistepScheduler":
         pipe.scheduler = UniPCMultistepScheduler.from_config(pipe.scheduler.config, flow_shift=flow_shift)
@@ -226,7 +222,7 @@ with gr.Blocks() as demo:
             
     generate_btn.click(
         fn=generate_video,
-        inputs=[ 
+        inputs=[
             model_choice,
             prompt,
             negative_prompt,
@@ -246,6 +242,12 @@ with gr.Blocks() as demo:
         outputs=[output_video, used_seed]
     )
     
-    gr.Markdown(""" ## Tips for best results: - Smaller videos: Flow shift 2.0–5.0 - Larger videos: Flow shift 7.0–12.0 - Use frame count in 4k+1 form (e.g., 33, 65) - Limit frame count and resolution to avoid timeout """)
+    gr.Markdown("""
+    ## Tips for best results:
+    - Smaller videos: Flow shift 2.0–5.0
+    - Larger videos: Flow shift 7.0–12.0
+    - Use frame count in 4k+1 form (e.g., 33, 65)
+    - Limit frame count and resolution to avoid timeout
+    """)
 
-demo.launch()
+demo.launch()