kulkas2pintu commited on
Commit ·
329fea3
1
Parent(s): 20d7733
Upgrade to WAMU_v3 + default 4 steps (faster)
Browse files
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
|
| 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=
|
| 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(
|