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
bac8a241
Unverified
Commit
bac8a241
authored
Sep 24, 2024
by
YiYi Xu
Committed by
GitHub
Sep 24, 2024
Browse files
a few fix for SingleFile tests (#9522)
* update sd15 repo * update more
parent
28f9d845
Changes
26
Hide whitespace changes
Inline
Side-by-side
Showing
6 changed files
with
18 additions
and
10 deletions
+18
-10
tests/single_file/test_stable_diffusion_controlnet_img2img_single_file.py
...e/test_stable_diffusion_controlnet_img2img_single_file.py
+4
-2
tests/single_file/test_stable_diffusion_controlnet_inpaint_single_file.py
...e/test_stable_diffusion_controlnet_inpaint_single_file.py
+1
-1
tests/single_file/test_stable_diffusion_controlnet_single_file.py
...ngle_file/test_stable_diffusion_controlnet_single_file.py
+4
-2
tests/single_file/test_stable_diffusion_img2img_single_file.py
.../single_file/test_stable_diffusion_img2img_single_file.py
+4
-2
tests/single_file/test_stable_diffusion_inpaint_single_file.py
.../single_file/test_stable_diffusion_inpaint_single_file.py
+1
-1
tests/single_file/test_stable_diffusion_single_file.py
tests/single_file/test_stable_diffusion_single_file.py
+4
-2
No files found.
tests/single_file/test_stable_diffusion_controlnet_img2img_single_file.py
View file @
bac8a241
...
@@ -30,11 +30,13 @@ enable_full_determinism()
...
@@ -30,11 +30,13 @@ enable_full_determinism()
@
require_torch_gpu
@
require_torch_gpu
class
StableDiffusionControlNetPipelineSingleFileSlowTests
(
unittest
.
TestCase
,
SDSingleFileTesterMixin
):
class
StableDiffusionControlNetPipelineSingleFileSlowTests
(
unittest
.
TestCase
,
SDSingleFileTesterMixin
):
pipeline_class
=
StableDiffusionControlNetPipeline
pipeline_class
=
StableDiffusionControlNetPipeline
ckpt_path
=
"https://huggingface.co/Jiali/stable-diffusion-1.5/blob/main/v1-5-pruned-emaonly.safetensors"
ckpt_path
=
(
"https://huggingface.co/stable-diffusion-v1-5/stable-diffusion-v1-5/blob/main/v1-5-pruned-emaonly.safetensors"
)
original_config
=
(
original_config
=
(
"https://raw.githubusercontent.com/CompVis/stable-diffusion/main/configs/stable-diffusion/v1-inference.yaml"
"https://raw.githubusercontent.com/CompVis/stable-diffusion/main/configs/stable-diffusion/v1-inference.yaml"
)
)
repo_id
=
"
Jiali
/stable-diffusion-
1.
5"
repo_id
=
"
stable-diffusion-v1-5
/stable-diffusion-
v1-
5"
def
setUp
(
self
):
def
setUp
(
self
):
super
().
setUp
()
super
().
setUp
()
...
...
tests/single_file/test_stable_diffusion_controlnet_inpaint_single_file.py
View file @
bac8a241
...
@@ -31,7 +31,7 @@ class StableDiffusionControlNetInpaintPipelineSingleFileSlowTests(unittest.TestC
...
@@ -31,7 +31,7 @@ class StableDiffusionControlNetInpaintPipelineSingleFileSlowTests(unittest.TestC
pipeline_class
=
StableDiffusionControlNetInpaintPipeline
pipeline_class
=
StableDiffusionControlNetInpaintPipeline
ckpt_path
=
"https://huggingface.co/botp/stable-diffusion-v1-5-inpainting/blob/main/sd-v1-5-inpainting.ckpt"
ckpt_path
=
"https://huggingface.co/botp/stable-diffusion-v1-5-inpainting/blob/main/sd-v1-5-inpainting.ckpt"
original_config
=
"https://raw.githubusercontent.com/runwayml/stable-diffusion/main/configs/stable-diffusion/v1-inpainting-inference.yaml"
original_config
=
"https://raw.githubusercontent.com/runwayml/stable-diffusion/main/configs/stable-diffusion/v1-inpainting-inference.yaml"
repo_id
=
"
botp/
stable-diffusion-v1-5-inpainting"
repo_id
=
"stable-diffusion-v1-5
/stable-diffusion
-inpainting"
def
setUp
(
self
):
def
setUp
(
self
):
super
().
setUp
()
super
().
setUp
()
...
...
tests/single_file/test_stable_diffusion_controlnet_single_file.py
View file @
bac8a241
...
@@ -29,11 +29,13 @@ enable_full_determinism()
...
@@ -29,11 +29,13 @@ enable_full_determinism()
@
require_torch_gpu
@
require_torch_gpu
class
StableDiffusionControlNetPipelineSingleFileSlowTests
(
unittest
.
TestCase
,
SDSingleFileTesterMixin
):
class
StableDiffusionControlNetPipelineSingleFileSlowTests
(
unittest
.
TestCase
,
SDSingleFileTesterMixin
):
pipeline_class
=
StableDiffusionControlNetPipeline
pipeline_class
=
StableDiffusionControlNetPipeline
ckpt_path
=
"https://huggingface.co/Jiali/stable-diffusion-1.5/blob/main/v1-5-pruned-emaonly.safetensors"
ckpt_path
=
(
"https://huggingface.co/stable-diffusion-v1-5/stable-diffusion-v1-5/blob/main/v1-5-pruned-emaonly.safetensors"
)
original_config
=
(
original_config
=
(
"https://raw.githubusercontent.com/CompVis/stable-diffusion/main/configs/stable-diffusion/v1-inference.yaml"
"https://raw.githubusercontent.com/CompVis/stable-diffusion/main/configs/stable-diffusion/v1-inference.yaml"
)
)
repo_id
=
"
Jiali
/stable-diffusion-
1.
5"
repo_id
=
"
stable-diffusion-v1-5
/stable-diffusion-
v1-
5"
def
setUp
(
self
):
def
setUp
(
self
):
super
().
setUp
()
super
().
setUp
()
...
...
tests/single_file/test_stable_diffusion_img2img_single_file.py
View file @
bac8a241
...
@@ -23,11 +23,13 @@ enable_full_determinism()
...
@@ -23,11 +23,13 @@ enable_full_determinism()
@
require_torch_gpu
@
require_torch_gpu
class
StableDiffusionImg2ImgPipelineSingleFileSlowTests
(
unittest
.
TestCase
,
SDSingleFileTesterMixin
):
class
StableDiffusionImg2ImgPipelineSingleFileSlowTests
(
unittest
.
TestCase
,
SDSingleFileTesterMixin
):
pipeline_class
=
StableDiffusionImg2ImgPipeline
pipeline_class
=
StableDiffusionImg2ImgPipeline
ckpt_path
=
"https://huggingface.co/Jiali/stable-diffusion-1.5/blob/main/v1-5-pruned-emaonly.safetensors"
ckpt_path
=
(
"https://huggingface.co/stable-diffusion-v1-5/stable-diffusion-v1-5/blob/main/v1-5-pruned-emaonly.safetensors"
)
original_config
=
(
original_config
=
(
"https://raw.githubusercontent.com/CompVis/stable-diffusion/main/configs/stable-diffusion/v1-inference.yaml"
"https://raw.githubusercontent.com/CompVis/stable-diffusion/main/configs/stable-diffusion/v1-inference.yaml"
)
)
repo_id
=
"
Jiali
/stable-diffusion-
1.
5"
repo_id
=
"
stable-diffusion-v1-5
/stable-diffusion-
v1-
5"
def
setUp
(
self
):
def
setUp
(
self
):
super
().
setUp
()
super
().
setUp
()
...
...
tests/single_file/test_stable_diffusion_inpaint_single_file.py
View file @
bac8a241
...
@@ -63,7 +63,7 @@ class StableDiffusionInpaintPipelineSingleFileSlowTests(unittest.TestCase, SDSin
...
@@ -63,7 +63,7 @@ class StableDiffusionInpaintPipelineSingleFileSlowTests(unittest.TestCase, SDSin
def
test_single_file_loading_4_channel_unet
(
self
):
def
test_single_file_loading_4_channel_unet
(
self
):
# Test loading single file inpaint with a 4 channel UNet
# Test loading single file inpaint with a 4 channel UNet
ckpt_path
=
"https://huggingface.co/
Jiali
/stable-diffusion-
1.
5/blob/main/v1-5-pruned-emaonly.safetensors"
ckpt_path
=
"https://huggingface.co/
stable-diffusion-v1-5
/stable-diffusion-
v1-
5/blob/main/v1-5-pruned-emaonly.safetensors"
pipe
=
self
.
pipeline_class
.
from_single_file
(
ckpt_path
)
pipe
=
self
.
pipeline_class
.
from_single_file
(
ckpt_path
)
assert
pipe
.
unet
.
config
.
in_channels
==
4
assert
pipe
.
unet
.
config
.
in_channels
==
4
...
...
tests/single_file/test_stable_diffusion_single_file.py
View file @
bac8a241
...
@@ -26,11 +26,13 @@ enable_full_determinism()
...
@@ -26,11 +26,13 @@ enable_full_determinism()
@
require_torch_gpu
@
require_torch_gpu
class
StableDiffusionPipelineSingleFileSlowTests
(
unittest
.
TestCase
,
SDSingleFileTesterMixin
):
class
StableDiffusionPipelineSingleFileSlowTests
(
unittest
.
TestCase
,
SDSingleFileTesterMixin
):
pipeline_class
=
StableDiffusionPipeline
pipeline_class
=
StableDiffusionPipeline
ckpt_path
=
"https://huggingface.co/Jiali/stable-diffusion-1.5/blob/main/v1-5-pruned-emaonly.safetensors"
ckpt_path
=
(
"https://huggingface.co/stable-diffusion-v1-5/stable-diffusion-v1-5/blob/main/v1-5-pruned-emaonly.safetensors"
)
original_config
=
(
original_config
=
(
"https://raw.githubusercontent.com/CompVis/stable-diffusion/main/configs/stable-diffusion/v1-inference.yaml"
"https://raw.githubusercontent.com/CompVis/stable-diffusion/main/configs/stable-diffusion/v1-inference.yaml"
)
)
repo_id
=
"
Jiali
/stable-diffusion-
1.
5"
repo_id
=
"
stable-diffusion-v1-5
/stable-diffusion-
v1-
5"
def
setUp
(
self
):
def
setUp
(
self
):
super
().
setUp
()
super
().
setUp
()
...
...
Prev
1
2
Next
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