ovi054 commited on
Commit
8ea00e2
·
verified ·
1 Parent(s): cc9b68f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -6
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-1.3B-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,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 = "Bright tones, overexposed, static, blurred details, subtitles, style, works, paintings, images, static, overall gray, worst quality, low quality, JPEG compression residue, ugly, incomplete, extra fingers, poorly drawn hands, poorly drawn faces, deformed, disfigured, misshapen limbs, fused fingers, still picture, messy background, three legs, many people in the background, walking backwards"),
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)