Spaces:
Running on Zero
Running on Zero
Choquinlabs commited on
Update app.py
Browse files
app.py
CHANGED
|
@@ -16,7 +16,7 @@ import numpy as np
|
|
| 16 |
import traceback
|
| 17 |
|
| 18 |
# ✅ NUEVO: importar optimización avanzada tipo Qwen-Image-MultipleAngles
|
| 19 |
-
from optimization import optimize_pipeline_
|
| 20 |
|
| 21 |
# Load LoRAs from JSON file
|
| 22 |
def load_loras_from_file():
|
|
@@ -62,6 +62,7 @@ pipe = DiffusionPipeline.from_pretrained(
|
|
| 62 |
base_model, scheduler=scheduler, torch_dtype=dtype
|
| 63 |
).to(device)
|
| 64 |
|
|
|
|
| 65 |
# ✅ NUEVO BLOQUE: aplicar AOT optimization (igual que Qwen-Image-MultipleAngles)
|
| 66 |
try:
|
| 67 |
example_args = (
|
|
@@ -78,6 +79,7 @@ try:
|
|
| 78 |
print("✅ Transformer AOT optimization complete.")
|
| 79 |
except Exception as e:
|
| 80 |
print(f"⚠️ AOT optimization skipped: {e}")
|
|
|
|
| 81 |
|
| 82 |
# Lightning LoRA info (no global state)
|
| 83 |
LIGHTNING_LORA_REPO = "lightx2v/Qwen-Image-Lightning"
|
|
|
|
| 16 |
import traceback
|
| 17 |
|
| 18 |
# ✅ NUEVO: importar optimización avanzada tipo Qwen-Image-MultipleAngles
|
| 19 |
+
#from optimization import optimize_pipeline_
|
| 20 |
|
| 21 |
# Load LoRAs from JSON file
|
| 22 |
def load_loras_from_file():
|
|
|
|
| 62 |
base_model, scheduler=scheduler, torch_dtype=dtype
|
| 63 |
).to(device)
|
| 64 |
|
| 65 |
+
"""
|
| 66 |
# ✅ NUEVO BLOQUE: aplicar AOT optimization (igual que Qwen-Image-MultipleAngles)
|
| 67 |
try:
|
| 68 |
example_args = (
|
|
|
|
| 79 |
print("✅ Transformer AOT optimization complete.")
|
| 80 |
except Exception as e:
|
| 81 |
print(f"⚠️ AOT optimization skipped: {e}")
|
| 82 |
+
"""
|
| 83 |
|
| 84 |
# Lightning LoRA info (no global state)
|
| 85 |
LIGHTNING_LORA_REPO = "lightx2v/Qwen-Image-Lightning"
|