Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
|
@@ -83,6 +83,7 @@ lora_path = hf_hub_download("gokaygokay/Flux-Game-Assets-LoRA-v2", "game_asst.sa
|
|
| 83 |
pipe.load_lora_weights(lora_path)
|
| 84 |
pipe.fuse_lora(lora_scale=1.0)
|
| 85 |
pipe.unload_lora_weights()
|
|
|
|
| 86 |
pipe.enable_model_cpu_offload()
|
| 87 |
|
| 88 |
|
|
|
|
| 83 |
pipe.load_lora_weights(lora_path)
|
| 84 |
pipe.fuse_lora(lora_scale=1.0)
|
| 85 |
pipe.unload_lora_weights()
|
| 86 |
+
pipe.to(device)
|
| 87 |
pipe.enable_model_cpu_offload()
|
| 88 |
|
| 89 |
|