blanchon commited on
Commit
0123799
·
verified ·
1 Parent(s): 6f3da65

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -0
app.py CHANGED
@@ -14,6 +14,9 @@ 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
  def calculate_optimal_dimensions(image: Image.Image):
19
  # Extract the original dimensions
 
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("blanchon/FluxFillFurniture", weight_name="lora_fill.safetensors")
18
+ pipe.fuse_lora(lora_scale=1.0)
19
+ pipe.to("cuda")
20
 
21
  def calculate_optimal_dimensions(image: Image.Image):
22
  # Extract the original dimensions