FLUX.1-Fill-dev

Running on Zero

multimodalart HF Staff commited on Nov 26, 2024

Commit

89f1f91

verified ·

1 Parent(s): c3fecad

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -13,7 +13,7 @@ from PIL import Image
 MAX_SEED = np.iinfo(np.int32).max
 MAX_IMAGE_SIZE = 2048
-pipe = FluxFillPipeline.from_pretrained("black-forest-labs/FLUX.1-Fill-dev", torch_dtype=torch.bfloat16, revision="refs/pr/4").to("cuda")
 def calculate_optimal_dimensions(image: Image.Image):
     # Extract the original dimensions

 MAX_SEED = np.iinfo(np.int32).max
 MAX_IMAGE_SIZE = 2048
+pipe = FluxFillPipeline.from_pretrained("black-forest-labs/FLUX.1-Fill-dev", torch_dtype=torch.bfloat16).to("cuda")
 def calculate_optimal_dimensions(image: Image.Image):
     # Extract the original dimensions