Spaces:
Running
on
Zero
Running
on
Zero
fix: type error
Browse files
app.py
CHANGED
|
@@ -340,8 +340,9 @@ def load_and_fuse_lora(lora_name):
|
|
| 340 |
|
| 341 |
load_and_fuse_additional_lora(lora_name)
|
| 342 |
|
| 343 |
-
# Ahead-of-time compilation with
|
| 344 |
-
|
|
|
|
| 345 |
|
| 346 |
print("Model compilation complete.")
|
| 347 |
|
|
|
|
| 340 |
|
| 341 |
load_and_fuse_additional_lora(lora_name)
|
| 342 |
|
| 343 |
+
# Ahead-of-time compilation with minimal memory footprint
|
| 344 |
+
# Use tiny images to minimize memory during compilation
|
| 345 |
+
optimize_pipeline_(pipe, image=[Image.new("RGB", (64, 64)), Image.new("RGB", (64, 64))], prompt="test")
|
| 346 |
|
| 347 |
print("Model compilation complete.")
|
| 348 |
|