dima0987654321 commited on
Commit
acebd80
·
verified ·
1 Parent(s): 650b977

Upload app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -19
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 Motion Enhancer and HMCumshot LoRAs into the
466
- # same transformer. They add their deltas on top of FlashGen.
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])