Diffusion-API

Runtime error

kadirnar commited on Mar 21, 2023

Commit

bf342c0

1 Parent(s): 3ceb86d

Update diffusion_webui/diffusion_models/controlnet/controlnet_seg.py

Files changed (1) hide show

diffusion_webui/diffusion_models/controlnet/controlnet_seg.py CHANGED Viewed

@@ -203,7 +203,7 @@ class StableDiffusionControlNetSegGenerator:
             "openmmlab/upernet-convnext-small"
         )
-        image = image_path["image"].convert("RGB").resize((512, 512))
         pixel_values = image_processor(image, return_tensors="pt").pixel_values
         with torch.no_grad():

             "openmmlab/upernet-convnext-small"
         )
+        image = Image.open(image_path).convert("RGB")
         pixel_values = image_processor(image, return_tensors="pt").pixel_values
         with torch.no_grad():