ovi054 commited on
Commit
ab613df
·
verified ·
1 Parent(s): 812e69e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -7,6 +7,8 @@ from transformers import T5EncoderModel, T5Tokenizer
7
  from PIL import Image
8
  import numpy as np
9
 
 
 
10
  model_id = "Wan-AI/Wan2.1-T2V-1.3B-Diffusers"
11
  vae = AutoencoderKLWan.from_pretrained(model_id, subfolder="vae", torch_dtype=torch.float32)
12
  pipe = WanPipeline.from_pretrained(model_id, vae=vae, torch_dtype=torch.bfloat16)
 
7
  from PIL import Image
8
  import numpy as np
9
 
10
+ import gradio as gr
11
+
12
  model_id = "Wan-AI/Wan2.1-T2V-1.3B-Diffusers"
13
  vae = AutoencoderKLWan.from_pretrained(model_id, subfolder="vae", torch_dtype=torch.float32)
14
  pipe = WanPipeline.from_pretrained(model_id, vae=vae, torch_dtype=torch.bfloat16)