lllyasviel
/

sd-controlnet-seg

stable-diffusion

Model card Files Files and versions

patrickvonplaten commited on Mar 8, 2023

Commit

48b3d1a

·

1 Parent(s): b966782

Update README.md

Files changed (1) hide show

README.md +2 -1

README.md CHANGED Viewed

@@ -85,11 +85,12 @@ import numpy as np
 from controlnet_utils import ade_palette
 import torch
 from diffusers import StableDiffusionControlNetPipeline, ControlNetModel, UniPCMultistepScheduler
 image_processor = AutoImageProcessor.from_pretrained("openmmlab/upernet-convnext-small")
 image_segmentor = UperNetForSemanticSegmentation.from_pretrained("openmmlab/upernet-convnext-small")
-image = Image.open("./images/house.png").convert('RGB')
 pixel_values = image_processor(image, return_tensors="pt").pixel_values

 from controlnet_utils import ade_palette
 import torch
 from diffusers import StableDiffusionControlNetPipeline, ControlNetModel, UniPCMultistepScheduler
+from diffusers.utils import load_image
 image_processor = AutoImageProcessor.from_pretrained("openmmlab/upernet-convnext-small")
 image_segmentor = UperNetForSemanticSegmentation.from_pretrained("openmmlab/upernet-convnext-small")
+image = load_image("https://huggingface.co/lllyasviel/sd-controlnet-seg/resolve/main/images/house.png").convert('RGB')
 pixel_values = image_processor(image, return_tensors="pt").pixel_values