flux-fill-outpaint

Runtime error

multimodalart HF staff commited on Oct 14

Commit

389369b

•

1 Parent(s): 6448b18

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -16,6 +16,7 @@ transformer = FluxTransformer2DModel.from_pretrained(
     "black-forest-labs/FLUX.1-dev", subfolder='transformer', torch_dtype=torch.bfloat16
 )
 vae = AutoencoderKL.from_pretrained("black-forest-labs/FLUX.1-dev", subfolder="vae").to("cuda")
 pipe = FluxControlNetInpaintingPipeline.from_pretrained(
     "black-forest-labs/FLUX.1-dev",
     controlnet=controlnet,

     "black-forest-labs/FLUX.1-dev", subfolder='transformer', torch_dtype=torch.bfloat16
 )
 vae = AutoencoderKL.from_pretrained("black-forest-labs/FLUX.1-dev", subfolder="vae").to("cuda")
+vae.to(torch.bfloat16)
 pipe = FluxControlNetInpaintingPipeline.from_pretrained(
     "black-forest-labs/FLUX.1-dev",
     controlnet=controlnet,