Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
@@ -10,7 +10,7 @@ import numpy as np
|
|
10 |
import gradio as gr
|
11 |
import spaces
|
12 |
|
13 |
-
model_id = "Wan-AI/Wan2.1-T2V-
|
14 |
vae = AutoencoderKLWan.from_pretrained(model_id, subfolder="vae", torch_dtype=torch.float32)
|
15 |
pipe = WanPipeline.from_pretrained(model_id, vae=vae, torch_dtype=torch.bfloat16)
|
16 |
flow_shift = 5.0 # 5.0 for 720P, 3.0 for 480P
|
@@ -37,13 +37,9 @@ iface = gr.Interface(
|
|
37 |
fn=generate,
|
38 |
inputs=[
|
39 |
gr.Textbox(label="Input prompt"),
|
40 |
-
# gr.Slider(label="Width", minimum=256, maximum=2048, step=8, value=1024),
|
41 |
-
# gr.Slider(label="Height", minimum=256, maximum=2048, step=8, value=1024),
|
42 |
-
# gr.Textbox(label="Lora ID", placeholder="Optional"),
|
43 |
-
# gr.Slider(minimum=0.0, maximum=1.0, step=0.01, label="Lora Scale", value=1)
|
44 |
],
|
45 |
additional_inputs = [
|
46 |
-
gr.Textbox(label="Negative prompt", value = "
|
47 |
gr.Slider(label="Width", minimum=480, maximum=1280, step=8, value=1280),
|
48 |
gr.Slider(label="Height", minimum=480, maximum=1280, step=8, value=720),
|
49 |
gr.Slider(minimum=20, maximum=100, step=1, label="Inference Steps", value=50)
|
|
|
10 |
import gradio as gr
|
11 |
import spaces
|
12 |
|
13 |
+
model_id = "Wan-AI/Wan2.1-T2V-14B-Diffusers"
|
14 |
vae = AutoencoderKLWan.from_pretrained(model_id, subfolder="vae", torch_dtype=torch.float32)
|
15 |
pipe = WanPipeline.from_pretrained(model_id, vae=vae, torch_dtype=torch.bfloat16)
|
16 |
flow_shift = 5.0 # 5.0 for 720P, 3.0 for 480P
|
|
|
37 |
fn=generate,
|
38 |
inputs=[
|
39 |
gr.Textbox(label="Input prompt"),
|
|
|
|
|
|
|
|
|
40 |
],
|
41 |
additional_inputs = [
|
42 |
+
gr.Textbox(label="Negative prompt", value = ""),
|
43 |
gr.Slider(label="Width", minimum=480, maximum=1280, step=8, value=1280),
|
44 |
gr.Slider(label="Height", minimum=480, maximum=1280, step=8, value=720),
|
45 |
gr.Slider(minimum=20, maximum=100, step=1, label="Inference Steps", value=50)
|