Update README.md
Browse files
README.md
CHANGED
|
@@ -56,7 +56,7 @@ controlnet_model = "InstantX/Qwen-Image-ControlNet-Inpainting"
|
|
| 56 |
|
| 57 |
controlnet = QwenImageControlNetModel.from_pretrained(controlnet_model, torch_dtype=torch.bfloat16)
|
| 58 |
|
| 59 |
-
pipe =
|
| 60 |
base_model, controlnet=controlnet, torch_dtype=torch.bfloat16
|
| 61 |
)
|
| 62 |
pipe.to("cuda")
|
|
|
|
| 56 |
|
| 57 |
controlnet = QwenImageControlNetModel.from_pretrained(controlnet_model, torch_dtype=torch.bfloat16)
|
| 58 |
|
| 59 |
+
pipe = QwenImageControlNetInpaintPipeline.from_pretrained(
|
| 60 |
base_model, controlnet=controlnet, torch_dtype=torch.bfloat16
|
| 61 |
)
|
| 62 |
pipe.to("cuda")
|