InstantX
/

Qwen-Image-ControlNet-Inpainting

QwenImageControlNetInpaintPipeline

Model card Files Files and versions

wanghaofan commited on 23 days ago

Commit

61fd4b9

·

verified ·

1 Parent(s): 81e59fc

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -56,7 +56,7 @@ controlnet_model = "InstantX/Qwen-Image-ControlNet-Inpainting"
 controlnet = QwenImageControlNetModel.from_pretrained(controlnet_model, torch_dtype=torch.bfloat16)
-pipe = QwenImageControlNetPipeline.from_pretrained(
     base_model, controlnet=controlnet, torch_dtype=torch.bfloat16
 )
 pipe.to("cuda")

 controlnet = QwenImageControlNetModel.from_pretrained(controlnet_model, torch_dtype=torch.bfloat16)
+pipe = QwenImageControlNetInpaintPipeline.from_pretrained(
     base_model, controlnet=controlnet, torch_dtype=torch.bfloat16
 )
 pipe.to("cuda")