rahul7star commited on
Commit
08b591b
Β·
verified Β·
1 Parent(s): 7ac9fcc

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -6,7 +6,9 @@ from diffusers import DiffusionPipeline
6
 
7
  from diffusers import DiffusionPipeline
8
 
9
- pipe = DiffusionPipeline.from_pretrained("StevenZhang/Wan2.1-T2V-1.3B-Diff")
 
 
10
 
11
  prompt = "Astronaut in a jungle, cold color palette, muted colors, detailed, 8k"
12
  image = pipe(prompt).images[0]
 
6
 
7
  from diffusers import DiffusionPipeline
8
 
9
+ from diffusers import DiffusionPipeline
10
+
11
+ pipe = DiffusionPipeline.from_pretrained("diffusers/Wan2.1-VAE")
12
 
13
  prompt = "Astronaut in a jungle, cold color palette, muted colors, detailed, 8k"
14
  image = pipe(prompt).images[0]