dima0987654321 commited on
Commit
8f0dfaf
Β·
verified Β·
1 Parent(s): 1acf164

Upload app.py

Browse files
Files changed (1) hide show
  1. app.py +19 -0
app.py CHANGED
@@ -90,6 +90,18 @@ SPATIAL_LORA_FILE = os.environ.get(
90
  )
91
  SPATIAL_LORA_SCALE = float(os.environ.get("H3_SPATIAL_LORA_SCALE", "0.4"))
92
 
 
 
 
 
 
 
 
 
 
 
 
 
93
  CONDITIONER_SPACE = os.environ.get("H3_CONDITIONER", "multimodalart/qwen3vl-conditioner")
94
  # ``pack`` places the transformer at startup (66 GB on disk via the spaces pack); VAEs move on first GPU call.
95
  PLACEMENT = os.environ.get("H3_PLACEMENT", "pack").lower()
@@ -529,6 +541,13 @@ def load_and_apply_extra_loras(transformer) -> list[str]:
529
  "Spatial & Physics LoRA",
530
  SPATIAL_LORA_SCALE,
531
  ),
 
 
 
 
 
 
 
532
  ]
533
 
534
  # ── Model loading ───────────────────────────────────────────────────────────
 
90
  )
91
  SPATIAL_LORA_SCALE = float(os.environ.get("H3_SPATIAL_LORA_SCALE", "0.4"))
92
 
93
+ # HMBreasts V2: H3 anatomy adapter for chest size/shape control.
94
+ # The weight is pulled directly from Hugging Face; there is no Civitai/API lookup.
95
+ BREASTS_LORA_REPO = os.environ.get(
96
+ "H3_BREASTS_LORA_REPO",
97
+ "Hearmeman/minimax-h3-loras",
98
+ )
99
+ BREASTS_LORA_FILE = os.environ.get(
100
+ "H3_BREASTS_LORA_FILE",
101
+ "HMBreastsV2.safetensors",
102
+ )
103
+ BREASTS_LORA_SCALE = float(os.environ.get("H3_BREASTS_LORA_SCALE", "1.0"))
104
+
105
  CONDITIONER_SPACE = os.environ.get("H3_CONDITIONER", "multimodalart/qwen3vl-conditioner")
106
  # ``pack`` places the transformer at startup (66 GB on disk via the spaces pack); VAEs move on first GPU call.
107
  PLACEMENT = os.environ.get("H3_PLACEMENT", "pack").lower()
 
541
  "Spatial & Physics LoRA",
542
  SPATIAL_LORA_SCALE,
543
  ),
544
+ _apply_extra_lora(
545
+ transformer,
546
+ BREASTS_LORA_REPO,
547
+ BREASTS_LORA_FILE,
548
+ "HMBreasts V2 LoRA",
549
+ BREASTS_LORA_SCALE,
550
+ ),
551
  ]
552
 
553
  # ── Model loading ───────────────────────────────────────────────────────────