ovi054 commited on
Commit
15e26e4
·
verified ·
1 Parent(s): 2aad192

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +0 -16
app.py CHANGED
@@ -16,22 +16,6 @@ model_id = "Wan-AI/Wan2.1-T2V-1.3B-Diffusers"
16
  vae = AutoencoderKLWan.from_pretrained(model_id, subfolder="vae", torch_dtype=torch.float32)
17
  pipe = WanPipeline.from_pretrained(model_id, vae=vae, torch_dtype=torch.bfloat16)
18
  flow_shift = 1.0 #5.0 1.0 for image, 5.0 for 720P, 3.0 for 480P
19
- # pipe.scheduler = UniPCMultistepScheduler.from_config(pipe.scheduler.config, flow_shift=flow_shift)
20
- import torch
21
- from diffusers import DDIMScheduler # Changed to DDIMScheduler
22
- from diffusers import WanPipeline, AutoencoderKLWan
23
- from para_attn.first_block_cache.diffusers_adapters import apply_cache_on_pipe
24
- from transformers import T5EncoderModel, T5Tokenizer
25
- from PIL import Image
26
- import numpy as np
27
- import gradio as gr
28
- import spaces
29
-
30
- model_id = "Wan-AI/Wan2.1-T2V-1.3B-Diffusers"
31
- vae = AutoencoderKLWan.from_pretrained(model_id, subfolder="vae", torch_dtype=torch.float32)
32
- pipe = WanPipeline.from_pretrained(model_id, vae=vae, torch_dtype=torch.bfloat16)
33
- flow_shift = 1.0 # 1.0 for image, 5.0 for 720P, 3.0 for 480P
34
-
35
  # pipe.scheduler = UniPCMultistepScheduler.from_config(pipe.scheduler.config, flow_shift=flow_shift)
36
 
37
  # Configure DDIMScheduler with a beta schedule
 
16
  vae = AutoencoderKLWan.from_pretrained(model_id, subfolder="vae", torch_dtype=torch.float32)
17
  pipe = WanPipeline.from_pretrained(model_id, vae=vae, torch_dtype=torch.bfloat16)
18
  flow_shift = 1.0 #5.0 1.0 for image, 5.0 for 720P, 3.0 for 480P
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
19
  # pipe.scheduler = UniPCMultistepScheduler.from_config(pipe.scheduler.config, flow_shift=flow_shift)
20
 
21
  # Configure DDIMScheduler with a beta schedule