Spaces:
Running on Zero
Running on Zero
Upload app.py
Browse files
app.py
CHANGED
|
@@ -42,15 +42,6 @@ LORA_REPO = os.environ.get("H3_LORA_REPO", "Beidouqixing/minimax-h3-4step-lora-f
|
|
| 42 |
LORA_FILE = os.environ.get("H3_LORA_FILE", "minimax_h3_4step_lora_flashgen_v1.0_768p_bf16.safetensors")
|
| 43 |
|
| 44 |
# Additional LoRAs, merged on top of the FlashGen 4-step weights.
|
| 45 |
-
MOTION_LORA_REPO = os.environ.get(
|
| 46 |
-
"H3_MOTION_LORA_REPO",
|
| 47 |
-
"rzgar/minimax-h3_fl2v_8Step_motion_enhancer",
|
| 48 |
-
)
|
| 49 |
-
MOTION_LORA_FILE = os.environ.get(
|
| 50 |
-
"H3_MOTION_LORA_FILE",
|
| 51 |
-
"minimax-h3_fl2v_8Step_motion_enhancer.safetensors",
|
| 52 |
-
)
|
| 53 |
-
MOTION_LORA_SCALE = float(os.environ.get("H3_MOTION_LORA_SCALE", "1.0"))
|
| 54 |
|
| 55 |
HMCUMSHOT_REPO = os.environ.get("H3_HMCUMSHOT_REPO", "Hearmeman/minimax-h3-loras")
|
| 56 |
HMCUMSHOT_FILE = os.environ.get("H3_HMCUMSHOT_FILE", "HMCumshot_V1.0.safetensors")
|
|
@@ -414,13 +405,6 @@ def _apply_extra_lora(
|
|
| 414 |
|
| 415 |
def load_and_apply_extra_loras(transformer) -> list[str]:
|
| 416 |
return [
|
| 417 |
-
_apply_extra_lora(
|
| 418 |
-
transformer,
|
| 419 |
-
MOTION_LORA_REPO,
|
| 420 |
-
MOTION_LORA_FILE,
|
| 421 |
-
"Motion Enhancer LoRA",
|
| 422 |
-
MOTION_LORA_SCALE,
|
| 423 |
-
),
|
| 424 |
_apply_extra_lora(
|
| 425 |
transformer,
|
| 426 |
HMCUMSHOT_REPO,
|
|
@@ -462,8 +446,8 @@ def load_models() -> str | None:
|
|
| 462 |
lora_status = load_and_apply_lora(pipe.transformer)
|
| 463 |
print(f"[gen] {lora_status}", flush=True)
|
| 464 |
|
| 465 |
-
# Then fold the requested
|
| 466 |
-
#
|
| 467 |
for extra_status in load_and_apply_extra_loras(pipe.transformer):
|
| 468 |
print(f"[gen] {extra_status}", flush=True)
|
| 469 |
|
|
@@ -874,4 +858,4 @@ with gr.Blocks(title="MiniMax-H3 4-Step FlashGen") as demo:
|
|
| 874 |
load_models()
|
| 875 |
|
| 876 |
if __name__ == "__main__":
|
| 877 |
-
demo.launch(show_error=True, theme=gr.themes.Citrus(), css=CSS, allowed_paths=[OUTPUT_DIR])
|
|
|
|
| 42 |
LORA_FILE = os.environ.get("H3_LORA_FILE", "minimax_h3_4step_lora_flashgen_v1.0_768p_bf16.safetensors")
|
| 43 |
|
| 44 |
# Additional LoRAs, merged on top of the FlashGen 4-step weights.
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 45 |
|
| 46 |
HMCUMSHOT_REPO = os.environ.get("H3_HMCUMSHOT_REPO", "Hearmeman/minimax-h3-loras")
|
| 47 |
HMCUMSHOT_FILE = os.environ.get("H3_HMCUMSHOT_FILE", "HMCumshot_V1.0.safetensors")
|
|
|
|
| 405 |
|
| 406 |
def load_and_apply_extra_loras(transformer) -> list[str]:
|
| 407 |
return [
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 408 |
_apply_extra_lora(
|
| 409 |
transformer,
|
| 410 |
HMCUMSHOT_REPO,
|
|
|
|
| 446 |
lora_status = load_and_apply_lora(pipe.transformer)
|
| 447 |
print(f"[gen] {lora_status}", flush=True)
|
| 448 |
|
| 449 |
+
# Then fold the requested HMCumshot LoRA into the same transformer.
|
| 450 |
+
# It adds its delta on top of FlashGen.
|
| 451 |
for extra_status in load_and_apply_extra_loras(pipe.transformer):
|
| 452 |
print(f"[gen] {extra_status}", flush=True)
|
| 453 |
|
|
|
|
| 858 |
load_models()
|
| 859 |
|
| 860 |
if __name__ == "__main__":
|
| 861 |
+
demo.launch(show_error=True, theme=gr.themes.Citrus(), css=CSS, allowed_paths=[OUTPUT_DIR])
|