YiYiXu commited on
Commit
ece023b
1 Parent(s): 95d384c

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +3 -2
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", subfolder="finetuned-from-20000-on-1024-checkpoint-24000", torch_dtype=torch.float16
 
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
- image[0]_.save(f"hug_lab.png")
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)