Daemontatox commited on
Commit
6b5f6ed
·
verified ·
1 Parent(s): f9cd25e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -1,3 +1,4 @@
 
1
  import torch
2
  from diffusers import UniPCMultistepScheduler
3
  from diffusers import WanPipeline, AutoencoderKLWan # Use Wan-specific VAE
@@ -8,7 +9,7 @@ from PIL import Image
8
  import numpy as np
9
 
10
  import gradio as gr
11
- import spaces
12
 
13
  model_id = "Wan-AI/Wan2.1-T2V-1.3B-Diffusers"
14
  vae = AutoencoderKLWan.from_pretrained(model_id, subfolder="vae", torch_dtype=torch.float32)
 
1
+ import spaces
2
  import torch
3
  from diffusers import UniPCMultistepScheduler
4
  from diffusers import WanPipeline, AutoencoderKLWan # Use Wan-specific VAE
 
9
  import numpy as np
10
 
11
  import gradio as gr
12
+
13
 
14
  model_id = "Wan-AI/Wan2.1-T2V-1.3B-Diffusers"
15
  vae = AutoencoderKLWan.from_pretrained(model_id, subfolder="vae", torch_dtype=torch.float32)