Update README.md
Browse files
README.md
CHANGED
|
@@ -24,6 +24,7 @@ import torch
|
|
| 24 |
from diffusers import StableDiffusion3ControlNetPipeline
|
| 25 |
from diffusers.models import SD3ControlNetModel, SD3MultiControlNetModel
|
| 26 |
from diffusers.utils import load_image
|
|
|
|
| 27 |
# load pipeline
|
| 28 |
controlnet = SD3ControlNetModel.from_pretrained("InstantX/SD3-Controlnet-Canny")
|
| 29 |
pipe = StableDiffusion3ControlNetPipeline.from_pretrained(
|
|
@@ -31,6 +32,7 @@ pipe = StableDiffusion3ControlNetPipeline.from_pretrained(
|
|
| 31 |
controlnet=controlnet
|
| 32 |
)
|
| 33 |
pipe.to("cuda", torch.float16)
|
|
|
|
| 34 |
# config
|
| 35 |
control_image = load_image("https://huggingface.co/InstantX/SD3-Controlnet-Canny/resolve/main/canny.jpg")
|
| 36 |
prompt = 'Anime style illustration of a girl wearing a suit. A moon in sky. In the background we see a big rain approaching. text "InstantX" on image'
|
|
|
|
| 24 |
from diffusers import StableDiffusion3ControlNetPipeline
|
| 25 |
from diffusers.models import SD3ControlNetModel, SD3MultiControlNetModel
|
| 26 |
from diffusers.utils import load_image
|
| 27 |
+
|
| 28 |
# load pipeline
|
| 29 |
controlnet = SD3ControlNetModel.from_pretrained("InstantX/SD3-Controlnet-Canny")
|
| 30 |
pipe = StableDiffusion3ControlNetPipeline.from_pretrained(
|
|
|
|
| 32 |
controlnet=controlnet
|
| 33 |
)
|
| 34 |
pipe.to("cuda", torch.float16)
|
| 35 |
+
|
| 36 |
# config
|
| 37 |
control_image = load_image("https://huggingface.co/InstantX/SD3-Controlnet-Canny/resolve/main/canny.jpg")
|
| 38 |
prompt = 'Anime style illustration of a girl wearing a suit. A moon in sky. In the background we see a big rain approaching. text "InstantX" on image'
|