Spaces:
Running
on
Zero
Running
on
Zero
Himanshu806
commited on
Update app.py
Browse files
app.py
CHANGED
@@ -14,10 +14,10 @@ MAX_SEED = np.iinfo(np.int32).max
|
|
14 |
MAX_IMAGE_SIZE = 2048
|
15 |
|
16 |
pipe = FluxFillPipeline.from_pretrained("black-forest-labs/FLUX.1-Fill-dev", torch_dtype=torch.bfloat16).to("cuda")
|
17 |
-
|
18 |
# pipe.enable_sequential_cpu_offload()
|
19 |
# pipe.enable_fp16()
|
20 |
-
|
21 |
# pipe.vae.enable_slicing()
|
22 |
# pipe.vae.enable_tiling()
|
23 |
|
|
|
14 |
MAX_IMAGE_SIZE = 2048
|
15 |
|
16 |
pipe = FluxFillPipeline.from_pretrained("black-forest-labs/FLUX.1-Fill-dev", torch_dtype=torch.bfloat16).to("cuda")
|
17 |
+
pipe.load_lora_weights("alvdansen/flux-koda")
|
18 |
# pipe.enable_sequential_cpu_offload()
|
19 |
# pipe.enable_fp16()
|
20 |
+
pipe.enable_lora()
|
21 |
# pipe.vae.enable_slicing()
|
22 |
# pipe.vae.enable_tiling()
|
23 |
|