Spaces:
Paused
Paused
Update app_lora1.py
Browse files- app_lora1.py +6 -1
app_lora1.py
CHANGED
|
@@ -92,7 +92,7 @@ def register_scripts(selected_scripts):
|
|
| 92 |
# GPU-ONLY PIPELINE BUILDER (CRITICAL)
|
| 93 |
# =========================================================
|
| 94 |
|
| 95 |
-
|
| 96 |
def get_pipeline(script_name):
|
| 97 |
if script_name in PIPELINES:
|
| 98 |
return PIPELINES[script_name]
|
|
@@ -164,6 +164,11 @@ def generate_image(
|
|
| 164 |
raise RuntimeError("Pipeline not registered")
|
| 165 |
|
| 166 |
pipe = get_pipeline(pipeline_name)
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 167 |
|
| 168 |
log("=== NEW GENERATION REQUEST ===")
|
| 169 |
log(f"Pipeline: {pipeline_name}")
|
|
|
|
| 92 |
# GPU-ONLY PIPELINE BUILDER (CRITICAL)
|
| 93 |
# =========================================================
|
| 94 |
|
| 95 |
+
|
| 96 |
def get_pipeline(script_name):
|
| 97 |
if script_name in PIPELINES:
|
| 98 |
return PIPELINES[script_name]
|
|
|
|
| 164 |
raise RuntimeError("Pipeline not registered")
|
| 165 |
|
| 166 |
pipe = get_pipeline(pipeline_name)
|
| 167 |
+
|
| 168 |
+
|
| 169 |
+
|
| 170 |
+
pipe = pipe.to("cuda")
|
| 171 |
+
|
| 172 |
|
| 173 |
log("=== NEW GENERATION REQUEST ===")
|
| 174 |
log(f"Pipeline: {pipeline_name}")
|