Spaces:
Running
on
Zero
Running
on
Zero
Update app_lora1.py
Browse files- app_lora1.py +3 -1
app_lora1.py
CHANGED
|
@@ -46,7 +46,8 @@ def pipeline_technology_info(pipe):
|
|
| 46 |
# Transformer dtype
|
| 47 |
if hasattr(pipe, "transformer"):
|
| 48 |
try:
|
| 49 |
-
tech.append(f"Transformer
|
|
|
|
| 50 |
except Exception:
|
| 51 |
pass
|
| 52 |
|
|
@@ -276,6 +277,7 @@ def generate_image(
|
|
| 276 |
):
|
| 277 |
log_buffer.truncate(0)
|
| 278 |
log_buffer.seek(0)
|
|
|
|
| 279 |
|
| 280 |
if pipeline_name not in SCRIPT_CODE:
|
| 281 |
raise RuntimeError("Pipeline not registered")
|
|
|
|
| 46 |
# Transformer dtype
|
| 47 |
if hasattr(pipe, "transformer"):
|
| 48 |
try:
|
| 49 |
+
tech.append(f"Transformer full info:\n{pipe.transformer}")
|
| 50 |
+
|
| 51 |
except Exception:
|
| 52 |
pass
|
| 53 |
|
|
|
|
| 277 |
):
|
| 278 |
log_buffer.truncate(0)
|
| 279 |
log_buffer.seek(0)
|
| 280 |
+
print(prompt)
|
| 281 |
|
| 282 |
if pipeline_name not in SCRIPT_CODE:
|
| 283 |
raise RuntimeError("Pipeline not registered")
|