kulkas2pintu commited on
Commit
329fea3
·
1 Parent(s): 20d7733

Upgrade to WAMU_v3 + default 4 steps (faster)

Browse files
Files changed (1) hide show
  1. app.py +2 -4
app.py CHANGED
@@ -234,9 +234,7 @@ def interpolate_bits(frames_np, multiplier=2, scale=1.0):
234
 
235
  ORG_NAME = "TestOrganizationPleaseIgnore"
236
  # MODEL_ID = "Wan-AI/Wan2.2-I2V-A14B-Diffusers"
237
- MODEL_ID = os.getenv("REPO_ID") or random.choice(
238
- list(list_models(author=ORG_NAME, filter='diffusers:WanImageToVideoPipeline'))
239
- ).modelId
240
  CACHE_DIR = os.path.expanduser("~/.cache/huggingface/")
241
 
242
  LORA_MODELS = [
@@ -662,7 +660,7 @@ with gr.Blocks(delete_cache=(3600, 10800)) as demo:
662
  quality_slider = gr.Slider(minimum=1, maximum=10, step=1, value=6, label="Video Quality", info="If set to 10, the generated video may be too large and won't play in the Gradio preview.")
663
  seed_input = gr.Slider(label="Seed", minimum=0, maximum=MAX_SEED, step=1, value=42, interactive=True)
664
  randomize_seed_checkbox = gr.Checkbox(label="Randomize seed", value=True, interactive=True)
665
- steps_slider = gr.Slider(minimum=1, maximum=30, step=1, value=6, label="Inference Steps")
666
  guidance_scale_input = gr.Slider(minimum=0.0, maximum=10.0, step=0.5, value=1, label="Guidance Scale - high noise stage", info="Values above 1 increase GPU usage and may take longer to process.")
667
  guidance_scale_2_input = gr.Slider(minimum=0.0, maximum=10.0, step=0.5, value=1, label="Guidance Scale 2 - low noise stage")
668
  scheduler_dropdown = gr.Dropdown(
 
234
 
235
  ORG_NAME = "TestOrganizationPleaseIgnore"
236
  # MODEL_ID = "Wan-AI/Wan2.2-I2V-A14B-Diffusers"
237
+ MODEL_ID = os.getenv("REPO_ID") or "TestOrganizationPleaseIgnore/WAMU_v3_WAN2.2_I2V_LIGHTNING"
 
 
238
  CACHE_DIR = os.path.expanduser("~/.cache/huggingface/")
239
 
240
  LORA_MODELS = [
 
660
  quality_slider = gr.Slider(minimum=1, maximum=10, step=1, value=6, label="Video Quality", info="If set to 10, the generated video may be too large and won't play in the Gradio preview.")
661
  seed_input = gr.Slider(label="Seed", minimum=0, maximum=MAX_SEED, step=1, value=42, interactive=True)
662
  randomize_seed_checkbox = gr.Checkbox(label="Randomize seed", value=True, interactive=True)
663
+ steps_slider = gr.Slider(minimum=1, maximum=30, step=1, value=4, label="Inference Steps", info="4 steps = fastest (Lightning default). Raise to 6-8 for slightly more detail.")
664
  guidance_scale_input = gr.Slider(minimum=0.0, maximum=10.0, step=0.5, value=1, label="Guidance Scale - high noise stage", info="Values above 1 increase GPU usage and may take longer to process.")
665
  guidance_scale_2_input = gr.Slider(minimum=0.0, maximum=10.0, step=0.5, value=1, label="Guidance Scale 2 - low noise stage")
666
  scheduler_dropdown = gr.Dropdown(