Spaces:
Runtime error
Runtime error
Revert "change flux model to lite one"
Browse filesThis reverts commit 694332de1933f8914e8412201b7b9dd2dfaa47f3.
app.py
CHANGED
|
@@ -65,12 +65,12 @@ pipe = FluxPipeline.from_pretrained(
|
|
| 65 |
)
|
| 66 |
log(f"Загрузка модели FLUX.1-dev завершена за {time.time() - start_time:.2f} секунд")
|
| 67 |
|
| 68 |
-
|
| 69 |
-
|
| 70 |
-
|
| 71 |
-
|
| 72 |
-
|
| 73 |
-
|
| 74 |
pipe.to(device)
|
| 75 |
|
| 76 |
start_time = time.time()
|
|
|
|
| 65 |
)
|
| 66 |
log(f"Загрузка модели FLUX.1-dev завершена за {time.time() - start_time:.2f} секунд")
|
| 67 |
|
| 68 |
+
start_time = time.time()
|
| 69 |
+
log(f"Загрузка LoRA")
|
| 70 |
+
pipe.load_lora_weights("Shakker-Labs/FLUX.1-dev-LoRA-add-details", weight_name="FLUX-dev-lora-add_details.safetensors")
|
| 71 |
+
log(f"Загрузка LoRA завершена за {time.time() - start_time:.2f} секунд")
|
| 72 |
+
|
| 73 |
+
pipe.fuse_lora(lora_scale=0.5)
|
| 74 |
pipe.to(device)
|
| 75 |
|
| 76 |
start_time = time.time()
|