Update README.md
Browse files
README.md
CHANGED
@@ -48,7 +48,8 @@ image = load_image("https://huggingface.co/datasets/hf-internal-testing/diffuser
|
|
48 |
controlnet_conditioning_scale = 0.5 # recommended for good generalization
|
49 |
|
50 |
controlnet = ControlNetModel.from_pretrained(
|
51 |
-
"diffusers/controlnet-sdxl-1.0",
|
|
|
52 |
)
|
53 |
vae = AutoencoderKL.from_pretrained("madebyollin/sdxl-vae-fp16-fix", torch_dtype=torch.float16)
|
54 |
pipe = StableDiffusionXLControlNetPipeline.from_pretrained(
|
@@ -68,7 +69,7 @@ images = pipe(
|
|
68 |
prompt, image=image, controlnet_conditioning_scale=controlnet_conditioning_scale,
|
69 |
).images
|
70 |
|
71 |
-
|
72 |
```
|
73 |
|
74 |
![images_10)](./out_hug_lab_7.png)
|
|
|
48 |
controlnet_conditioning_scale = 0.5 # recommended for good generalization
|
49 |
|
50 |
controlnet = ControlNetModel.from_pretrained(
|
51 |
+
"diffusers/controlnet-sdxl-1.0",
|
52 |
+
torch_dtype=torch.float16
|
53 |
)
|
54 |
vae = AutoencoderKL.from_pretrained("madebyollin/sdxl-vae-fp16-fix", torch_dtype=torch.float16)
|
55 |
pipe = StableDiffusionXLControlNetPipeline.from_pretrained(
|
|
|
69 |
prompt, image=image, controlnet_conditioning_scale=controlnet_conditioning_scale,
|
70 |
).images
|
71 |
|
72 |
+
images[0].save(f"hug_lab.png")
|
73 |
```
|
74 |
|
75 |
![images_10)](./out_hug_lab_7.png)
|