Spaces:

Himanshu806
/

fluxInpaint

Running on Zero

Himanshu-AT commited on 6 days ago

Commit

784ce74

1 Parent(s): c6f6d11

remove controlnet

Files changed (1) hide show

app.py CHANGED Viewed

@@ -16,10 +16,10 @@ transformer = FluxTransformer2DModel.from_pretrained(
     )
 # Build pipeline
-controlnet = FluxControlNetModel.from_pretrained("alimama-creative/FLUX.1-dev-Controlnet-Inpainting-Beta", torch_dtype=torch.bfloat16)
 pipe = FluxControlNetInpaintingPipeline.from_pretrained(
     "black-forest-labs/FLUX.1-dev",
-    controlnet=controlnet,
     transformer=transformer,
     torch_dtype=torch.bfloat16
 ).to("cuda")

     )
 # Build pipeline
+# controlnet = FluxControlNetModel.from_pretrained("alimama-creative/FLUX.1-dev-Controlnet-Inpainting-Beta", torch_dtype=torch.bfloat16)
 pipe = FluxControlNetInpaintingPipeline.from_pretrained(
     "black-forest-labs/FLUX.1-dev",
+    # controlnet=controlnet,
     transformer=transformer,
     torch_dtype=torch.bfloat16
 ).to("cuda")