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
OpenDAS
diffusers
Commits
bdecc3cf
Commit
bdecc3cf
authored
Jun 28, 2022
by
Patrick von Platen
Browse files
move pipelines into folders
parent
0efac0aa
Changes
24
Show whitespace changes
Inline
Side-by-side
Showing
4 changed files
with
6 additions
and
3 deletions
+6
-3
src/diffusers/pipelines/score_sde_ve/pipeline_score_sde_ve.py
...diffusers/pipelines/score_sde_ve/pipeline_score_sde_ve.py
+0
-0
src/diffusers/pipelines/score_sde_vp/__init__.py
src/diffusers/pipelines/score_sde_vp/__init__.py
+1
-0
src/diffusers/pipelines/score_sde_vp/pipeline_score_sde_vp.py
...diffusers/pipelines/score_sde_vp/pipeline_score_sde_vp.py
+0
-0
tests/test_modeling_utils.py
tests/test_modeling_utils.py
+5
-3
No files found.
src/diffusers/pipelines/pipeline_score_sde_ve.py
→
src/diffusers/pipelines/
score_sde_ve/
pipeline_score_sde_ve.py
View file @
bdecc3cf
File moved
src/diffusers/pipelines/score_sde_vp/__init__.py
0 → 100644
View file @
bdecc3cf
from
.pipeline_score_sde_vp
import
ScoreSdeVpPipeline
src/diffusers/pipelines/pipeline_score_sde_vp.py
→
src/diffusers/pipelines/
score_sde_vp/
pipeline_score_sde_vp.py
View file @
bdecc3cf
File moved
tests/test_modeling_utils.py
View file @
bdecc3cf
...
@@ -47,7 +47,7 @@ from diffusers import (
...
@@ -47,7 +47,7 @@ from diffusers import (
)
)
from
diffusers.configuration_utils
import
ConfigMixin
from
diffusers.configuration_utils
import
ConfigMixin
from
diffusers.pipeline_utils
import
DiffusionPipeline
from
diffusers.pipeline_utils
import
DiffusionPipeline
from
diffusers.pipelines.pipeline_bddm
import
DiffWave
from
diffusers.pipelines.
bddm.
pipeline_bddm
import
DiffWave
from
diffusers.testing_utils
import
floats_tensor
,
slow
,
torch_device
from
diffusers.testing_utils
import
floats_tensor
,
slow
,
torch_device
...
@@ -1005,11 +1005,13 @@ class PipelineTesterMixin(unittest.TestCase):
...
@@ -1005,11 +1005,13 @@ class PipelineTesterMixin(unittest.TestCase):
bddm
=
BDDMPipeline
(
model
,
noise_scheduler
)
bddm
=
BDDMPipeline
(
model
,
noise_scheduler
)
# check if the library name for the diffwave moduel is set to pipeline module
# check if the library name for the diffwave moduel is set to pipeline module
self
.
assertTrue
(
bddm
.
config
[
"diffwave"
][
0
]
==
"
pipeline_
bddm"
)
self
.
assertTrue
(
bddm
.
config
[
"diffwave"
][
0
]
==
"bddm"
)
# check if we can save and load the pipeline
# check if we can save and load the pipeline
with
tempfile
.
TemporaryDirectory
()
as
tmpdirname
:
with
tempfile
.
TemporaryDirectory
()
as
tmpdirname
:
bddm
.
save_pretrained
(
tmpdirname
)
bddm
.
save_pretrained
(
tmpdirname
)
_
=
BDDMPipeline
.
from_pretrained
(
tmpdirname
)
_
=
BDDMPipeline
.
from_pretrained
(
tmpdirname
)
# check if the same works using the DifusionPipeline class
# check if the same works using the DifusionPipeline class
_
=
DiffusionPipeline
.
from_pretrained
(
tmpdirname
)
bddm
=
DiffusionPipeline
.
from_pretrained
(
tmpdirname
)
self
.
assertTrue
(
bddm
.
config
[
"diffwave"
][
0
]
==
"bddm"
)
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