Himanshu806 commited on
Commit
c6696f9
·
verified ·
1 Parent(s): 9335613

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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
- # 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
 
 
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