Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
@@ -13,11 +13,9 @@ from PIL import Image
|
|
13 |
MAX_SEED = np.iinfo(np.int32).max
|
14 |
MAX_IMAGE_SIZE = 2048
|
15 |
|
16 |
-
pipe = FluxFillPipeline.from_pretrained("
|
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 |
|
|
|
13 |
MAX_SEED = np.iinfo(np.int32).max
|
14 |
MAX_IMAGE_SIZE = 2048
|
15 |
|
16 |
+
pipe = FluxFillPipeline.from_pretrained("Himanshu806/FluxHyperReal", torch_dtype=torch.bfloat16).to("cuda")
|
|
|
17 |
# pipe.enable_sequential_cpu_offload()
|
18 |
# pipe.enable_fp16()
|
|
|
19 |
# pipe.vae.enable_slicing()
|
20 |
# pipe.vae.enable_tiling()
|
21 |
|