Update app.py
Browse files
app.py
CHANGED
|
@@ -507,7 +507,27 @@ def generate_video(
|
|
| 507 |
except Exception as e:
|
| 508 |
print(f"[LoRA] pipeline rebuild FAILED: {e}")
|
| 509 |
|
| 510 |
-
#
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 511 |
log_memory("before pipeline call")
|
| 512 |
|
| 513 |
video, audio = pipeline(
|
|
|
|
| 507 |
except Exception as e:
|
| 508 |
print(f"[LoRA] pipeline rebuild FAILED: {e}")
|
| 509 |
|
| 510 |
+
# Force transformer rebuild so LoRA strength actually applies
|
| 511 |
+
try:
|
| 512 |
+
if hasattr(pipeline, "model_ledger"):
|
| 513 |
+
# Hard reset ALL known transformer references
|
| 514 |
+
if hasattr(pipeline.model_ledger, "_transformer"):
|
| 515 |
+
pipeline.model_ledger._transformer = None
|
| 516 |
+
|
| 517 |
+
# VERY IMPORTANT: also clear pipeline components cache if present
|
| 518 |
+
if hasattr(pipeline, "pipeline_components"):
|
| 519 |
+
try:
|
| 520 |
+
pipeline.pipeline_components = None
|
| 521 |
+
except Exception:
|
| 522 |
+
pass
|
| 523 |
+
|
| 524 |
+
# Force rebuild NOW (so it uses the new LoRA strength)
|
| 525 |
+
_ = pipeline.model_ledger.transformer()
|
| 526 |
+
|
| 527 |
+
print("[LoRA] transformer force-rebuilt with new strength")
|
| 528 |
+
except Exception as e:
|
| 529 |
+
print(f"[LoRA] transformer rebuild failed: {e}")
|
| 530 |
+
|
| 531 |
log_memory("before pipeline call")
|
| 532 |
|
| 533 |
video, audio = pipeline(
|