Update text.py
Browse files
text.py
CHANGED
|
@@ -51,6 +51,18 @@ upsample_pipe = LTX2LatentUpsamplePipeline(
|
|
| 51 |
latent_upsampler=latent_upsampler
|
| 52 |
).to(DEVICE)
|
| 53 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 54 |
print("✅ Model Loaded")
|
| 55 |
|
| 56 |
# ============================================================
|
|
|
|
| 51 |
latent_upsampler=latent_upsampler
|
| 52 |
).to(DEVICE)
|
| 53 |
|
| 54 |
+
|
| 55 |
+
pipe.load_lora_weights(
|
| 56 |
+
"Lightricks/LTX-2-19b-IC-LoRA-Detailer",
|
| 57 |
+
adapter_name="camera_control"
|
| 58 |
+
)
|
| 59 |
+
|
| 60 |
+
|
| 61 |
+
|
| 62 |
+
pipe.fuse_lora(lora_scale=0.8)
|
| 63 |
+
pipe.unload_lora_weights()
|
| 64 |
+
|
| 65 |
+
print("🔥 Model fully loaded on CUDA.")
|
| 66 |
print("✅ Model Loaded")
|
| 67 |
|
| 68 |
# ============================================================
|