Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
renzhc
diffusers_dcu
Commits
6bceaea3
Unverified
Commit
6bceaea3
authored
Feb 20, 2024
by
Dhruv Nair
Committed by
GitHub
Feb 20, 2024
Browse files
Update ControlNet Inpaint single file test (#7022)
update
parent
baf9924b
Changes
1
Show whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
2 additions
and
3 deletions
+2
-3
tests/pipelines/controlnet/test_controlnet_inpaint.py
tests/pipelines/controlnet/test_controlnet_inpaint.py
+2
-3
No files found.
tests/pipelines/controlnet/test_controlnet_inpaint.py
View file @
6bceaea3
...
...
@@ -559,17 +559,16 @@ class ControlNetInpaintPipelineSlowTests(unittest.TestCase):
def
test_load_local
(
self
):
controlnet
=
ControlNetModel
.
from_pretrained
(
"lllyasviel/control_v11p_sd15_canny"
)
pipe_1
=
StableDiffusionControlNetInpaintPipeline
.
from_pretrained
(
"runwayml/stable-diffusion-
v1-5
"
,
safety_checker
=
None
,
controlnet
=
controlnet
"runwayml/stable-diffusion-
inpainting
"
,
safety_checker
=
None
,
controlnet
=
controlnet
)
controlnet
=
ControlNetModel
.
from_single_file
(
"https://huggingface.co/lllyasviel/ControlNet-v1-1/blob/main/control_v11p_sd15_canny.pth"
)
pipe_2
=
StableDiffusionControlNetInpaintPipeline
.
from_single_file
(
"https://huggingface.co/runwayml/stable-diffusion-
v1-5
/blob/main/v1-5-
pruned-emaonly.safetensors
"
,
"https://huggingface.co/runwayml/stable-diffusion-
inpainting
/blob/main/
sd-
v1-5-
inpainting.ckpt
"
,
safety_checker
=
None
,
controlnet
=
controlnet
,
scheduler_type
=
"pndm"
,
)
control_image
=
load_image
(
"https://huggingface.co/datasets/hf-internal-testing/diffusers-images/resolve/main/sd_controlnet/bird_canny.png"
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment