Update README.md
Browse files
README.md
CHANGED
|
@@ -33,13 +33,13 @@ pipe.load_components(torch_dtype=torch.bfloat16)
|
|
| 33 |
controlnet_spec = pipeline.get_component_spec("controlnet")
|
| 34 |
controlnet_spec.pretrained_model_name_or_path = "InstantX/Qwen-Image-ControlNet-Union"
|
| 35 |
controlnet = controlnet_spec.load(torch_dtype=torch.bfloat16)
|
| 36 |
-
|
| 37 |
|
| 38 |
# Infer
|
| 39 |
prompt = "cat wizard with red hat, gandalf, lord of the rings, detailed, fantasy, cute, adorable, Pixar, Disney"
|
| 40 |
image = load_image("https://github.com/Trgtuan10/Image_storage/blob/main/cute_cat.png?raw=true")
|
| 41 |
|
| 42 |
-
output =
|
| 43 |
prompt=prompt,
|
| 44 |
image=image,
|
| 45 |
).images[0]
|
|
|
|
| 33 |
controlnet_spec = pipeline.get_component_spec("controlnet")
|
| 34 |
controlnet_spec.pretrained_model_name_or_path = "InstantX/Qwen-Image-ControlNet-Union"
|
| 35 |
controlnet = controlnet_spec.load(torch_dtype=torch.bfloat16)
|
| 36 |
+
pipe.update_components(controlnet=controlnet)
|
| 37 |
|
| 38 |
# Infer
|
| 39 |
prompt = "cat wizard with red hat, gandalf, lord of the rings, detailed, fantasy, cute, adorable, Pixar, Disney"
|
| 40 |
image = load_image("https://github.com/Trgtuan10/Image_storage/blob/main/cute_cat.png?raw=true")
|
| 41 |
|
| 42 |
+
output = pipe(
|
| 43 |
prompt=prompt,
|
| 44 |
image=image,
|
| 45 |
).images[0]
|