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
1d2587bb
Unverified
Commit
1d2587bb
authored
Aug 03, 2023
by
Dhruv Nair
Committed by
GitHub
Aug 03, 2023
Browse files
move tests to nightly (#4451)
* move tests to nightly * clean up code quality issues * more clean up
parent
372b5810
Changes
13
Show whitespace changes
Inline
Side-by-side
Showing
13 changed files
with
26 additions
and
26 deletions
+26
-26
tests/pipelines/altdiffusion/test_alt_diffusion.py
tests/pipelines/altdiffusion/test_alt_diffusion.py
+2
-2
tests/pipelines/altdiffusion/test_alt_diffusion_img2img.py
tests/pipelines/altdiffusion/test_alt_diffusion_img2img.py
+2
-2
tests/pipelines/audio_diffusion/test_audio_diffusion.py
tests/pipelines/audio_diffusion/test_audio_diffusion.py
+2
-2
tests/pipelines/consistency_models/test_consistency_models.py
...s/pipelines/consistency_models/test_consistency_models.py
+2
-2
tests/pipelines/dance_diffusion/test_dance_diffusion.py
tests/pipelines/dance_diffusion/test_dance_diffusion.py
+2
-2
tests/pipelines/dit/test_dit.py
tests/pipelines/dit/test_dit.py
+2
-2
tests/pipelines/karras_ve/test_karras_ve.py
tests/pipelines/karras_ve/test_karras_ve.py
+2
-2
tests/pipelines/paint_by_example/test_paint_by_example.py
tests/pipelines/paint_by_example/test_paint_by_example.py
+2
-2
tests/pipelines/pndm/test_pndm.py
tests/pipelines/pndm/test_pndm.py
+2
-2
tests/pipelines/score_sde_ve/test_score_sde_ve.py
tests/pipelines/score_sde_ve/test_score_sde_ve.py
+2
-2
tests/pipelines/spectrogram_diffusion/test_spectrogram_diffusion.py
...lines/spectrogram_diffusion/test_spectrogram_diffusion.py
+2
-2
tests/pipelines/versatile_diffusion/test_versatile_diffusion_image_variation.py
...ile_diffusion/test_versatile_diffusion_image_variation.py
+2
-2
tests/pipelines/vq_diffusion/test_vq_diffusion.py
tests/pipelines/vq_diffusion/test_vq_diffusion.py
+2
-2
No files found.
tests/pipelines/altdiffusion/test_alt_diffusion.py
View file @
1d2587bb
...
...
@@ -25,7 +25,7 @@ from diffusers.pipelines.alt_diffusion.modeling_roberta_series import (
RobertaSeriesConfig
,
RobertaSeriesModelWithTransformation
,
)
from
diffusers.utils
import
slow
,
torch_device
from
diffusers.utils
import
nightly
,
torch_device
from
diffusers.utils.testing_utils
import
enable_full_determinism
,
require_torch_gpu
from
..pipeline_params
import
TEXT_TO_IMAGE_BATCH_PARAMS
,
TEXT_TO_IMAGE_IMAGE_PARAMS
,
TEXT_TO_IMAGE_PARAMS
...
...
@@ -205,7 +205,7 @@ class AltDiffusionPipelineFastTests(
assert
np
.
abs
(
image_slice
.
flatten
()
-
expected_slice
).
max
()
<
1e-2
@
slow
@
nightly
@
require_torch_gpu
class
AltDiffusionPipelineIntegrationTests
(
unittest
.
TestCase
):
def
tearDown
(
self
):
...
...
tests/pipelines/altdiffusion/test_alt_diffusion_img2img.py
View file @
1d2587bb
...
...
@@ -32,7 +32,7 @@ from diffusers.pipelines.alt_diffusion.modeling_roberta_series import (
RobertaSeriesConfig
,
RobertaSeriesModelWithTransformation
,
)
from
diffusers.utils
import
floats_tensor
,
load_image
,
load_numpy
,
slow
,
torch_device
from
diffusers.utils
import
floats_tensor
,
load_image
,
load_numpy
,
nightly
,
torch_device
from
diffusers.utils.testing_utils
import
enable_full_determinism
,
require_torch_gpu
...
...
@@ -254,7 +254,7 @@ class AltDiffusionImg2ImgPipelineFastTests(unittest.TestCase):
assert
np
.
abs
(
image_slice
.
flatten
()
-
expected_slice
).
max
()
<
1e-2
@
slow
@
nightly
@
require_torch_gpu
class
AltDiffusionImg2ImgPipelineIntegrationTests
(
unittest
.
TestCase
):
def
tearDown
(
self
):
...
...
tests/pipelines/audio_diffusion/test_audio_diffusion.py
View file @
1d2587bb
...
...
@@ -29,7 +29,7 @@ from diffusers import (
UNet2DConditionModel
,
UNet2DModel
,
)
from
diffusers.utils
import
slow
,
torch_device
from
diffusers.utils
import
nightly
,
slow
,
torch_device
from
diffusers.utils.testing_utils
import
enable_full_determinism
,
require_torch_gpu
...
...
@@ -175,7 +175,7 @@ class PipelineFastTests(unittest.TestCase):
assert
np
.
abs
(
image_slice
.
flatten
()
-
expected_slice
).
max
()
==
0
@
slow
@
nightly
@
require_torch_gpu
class
PipelineIntegrationTests
(
unittest
.
TestCase
):
def
tearDown
(
self
):
...
...
tests/pipelines/consistency_models/test_consistency_models.py
View file @
1d2587bb
...
...
@@ -10,7 +10,7 @@ from diffusers import (
ConsistencyModelPipeline
,
UNet2DModel
,
)
from
diffusers.utils
import
randn_tensor
,
slow
,
torch_device
from
diffusers.utils
import
nightly
,
randn_tensor
,
torch_device
from
diffusers.utils.testing_utils
import
enable_full_determinism
,
require_torch_2
,
require_torch_gpu
from
..pipeline_params
import
UNCONDITIONAL_IMAGE_GENERATION_BATCH_PARAMS
,
UNCONDITIONAL_IMAGE_GENERATION_PARAMS
...
...
@@ -161,7 +161,7 @@ class ConsistencyModelPipelineFastTests(PipelineTesterMixin, unittest.TestCase):
assert
np
.
abs
(
image_slice
.
flatten
()
-
expected_slice
).
max
()
<
1e-3
@
slow
@
nightly
@
require_torch_gpu
class
ConsistencyModelPipelineSlowTests
(
unittest
.
TestCase
):
def
tearDown
(
self
):
...
...
tests/pipelines/dance_diffusion/test_dance_diffusion.py
View file @
1d2587bb
...
...
@@ -20,7 +20,7 @@ import numpy as np
import
torch
from
diffusers
import
DanceDiffusionPipeline
,
IPNDMScheduler
,
UNet1DModel
from
diffusers.utils
import
slow
,
torch_device
from
diffusers.utils
import
nightly
,
torch_device
from
diffusers.utils.testing_utils
import
enable_full_determinism
,
require_torch_gpu
,
skip_mps
from
..pipeline_params
import
UNCONDITIONAL_AUDIO_GENERATION_BATCH_PARAMS
,
UNCONDITIONAL_AUDIO_GENERATION_PARAMS
...
...
@@ -116,7 +116,7 @@ class DanceDiffusionPipelineFastTests(PipelineTesterMixin, unittest.TestCase):
super
().
test_inference_batch_single_identical
(
expected_max_diff
=
3e-3
)
@
slow
@
nightly
@
require_torch_gpu
class
PipelineIntegrationTests
(
unittest
.
TestCase
):
def
tearDown
(
self
):
...
...
tests/pipelines/dit/test_dit.py
View file @
1d2587bb
...
...
@@ -20,7 +20,7 @@ import numpy as np
import
torch
from
diffusers
import
AutoencoderKL
,
DDIMScheduler
,
DiTPipeline
,
DPMSolverMultistepScheduler
,
Transformer2DModel
from
diffusers.utils
import
is_xformers_available
,
load_numpy
,
slow
,
torch_device
from
diffusers.utils
import
is_xformers_available
,
load_numpy
,
nightly
,
torch_device
from
diffusers.utils.testing_utils
import
enable_full_determinism
,
require_torch_gpu
from
..pipeline_params
import
(
...
...
@@ -106,8 +106,8 @@ class DiTPipelineFastTests(PipelineTesterMixin, unittest.TestCase):
self
.
_test_xformers_attention_forwardGenerator_pass
(
expected_max_diff
=
1e-3
)
@
nightly
@
require_torch_gpu
@
slow
class
DiTPipelineIntegrationTests
(
unittest
.
TestCase
):
def
tearDown
(
self
):
super
().
tearDown
()
...
...
tests/pipelines/karras_ve/test_karras_ve.py
View file @
1d2587bb
...
...
@@ -19,7 +19,7 @@ import numpy as np
import
torch
from
diffusers
import
KarrasVePipeline
,
KarrasVeScheduler
,
UNet2DModel
from
diffusers.utils.testing_utils
import
enable_full_determinism
,
require_torch
,
slow
,
torch_device
from
diffusers.utils.testing_utils
import
enable_full_determinism
,
nightly
,
require_torch
,
torch_device
enable_full_determinism
()
...
...
@@ -64,7 +64,7 @@ class KarrasVePipelineFastTests(unittest.TestCase):
assert
np
.
abs
(
image_from_tuple_slice
.
flatten
()
-
expected_slice
).
max
()
<
1e-2
@
slow
@
nightly
@
require_torch
class
KarrasVePipelineIntegrationTests
(
unittest
.
TestCase
):
def
test_inference
(
self
):
...
...
tests/pipelines/paint_by_example/test_paint_by_example.py
View file @
1d2587bb
...
...
@@ -24,7 +24,7 @@ from transformers import CLIPImageProcessor, CLIPVisionConfig
from
diffusers
import
AutoencoderKL
,
PaintByExamplePipeline
,
PNDMScheduler
,
UNet2DConditionModel
from
diffusers.pipelines.paint_by_example
import
PaintByExampleImageEncoder
from
diffusers.utils
import
floats_tensor
,
load_image
,
slow
,
torch_device
from
diffusers.utils
import
floats_tensor
,
load_image
,
nightly
,
torch_device
from
diffusers.utils.testing_utils
import
enable_full_determinism
,
require_torch_gpu
from
..pipeline_params
import
IMAGE_GUIDED_IMAGE_INPAINTING_BATCH_PARAMS
,
IMAGE_GUIDED_IMAGE_INPAINTING_PARAMS
...
...
@@ -165,7 +165,7 @@ class PaintByExamplePipelineFastTests(PipelineTesterMixin, unittest.TestCase):
super
().
test_inference_batch_single_identical
(
expected_max_diff
=
3e-3
)
@
slow
@
nightly
@
require_torch_gpu
class
PaintByExamplePipelineIntegrationTests
(
unittest
.
TestCase
):
def
tearDown
(
self
):
...
...
tests/pipelines/pndm/test_pndm.py
View file @
1d2587bb
...
...
@@ -19,7 +19,7 @@ import numpy as np
import
torch
from
diffusers
import
PNDMPipeline
,
PNDMScheduler
,
UNet2DModel
from
diffusers.utils.testing_utils
import
enable_full_determinism
,
require_torch
,
slow
,
torch_device
from
diffusers.utils.testing_utils
import
enable_full_determinism
,
nightly
,
require_torch
,
torch_device
enable_full_determinism
()
...
...
@@ -64,7 +64,7 @@ class PNDMPipelineFastTests(unittest.TestCase):
assert
np
.
abs
(
image_from_tuple_slice
.
flatten
()
-
expected_slice
).
max
()
<
1e-2
@
slow
@
nightly
@
require_torch
class
PNDMPipelineIntegrationTests
(
unittest
.
TestCase
):
def
test_inference_cifar10
(
self
):
...
...
tests/pipelines/score_sde_ve/test_score_sde_ve.py
View file @
1d2587bb
...
...
@@ -19,7 +19,7 @@ import numpy as np
import
torch
from
diffusers
import
ScoreSdeVePipeline
,
ScoreSdeVeScheduler
,
UNet2DModel
from
diffusers.utils.testing_utils
import
enable_full_determinism
,
require_torch
,
slow
,
torch_device
from
diffusers.utils.testing_utils
import
enable_full_determinism
,
nightly
,
require_torch
,
torch_device
enable_full_determinism
()
...
...
@@ -66,7 +66,7 @@ class ScoreSdeVeipelineFastTests(unittest.TestCase):
assert
np
.
abs
(
image_from_tuple_slice
.
flatten
()
-
expected_slice
).
max
()
<
1e-2
@
slow
@
nightly
@
require_torch
class
ScoreSdeVePipelineIntegrationTests
(
unittest
.
TestCase
):
def
test_inference
(
self
):
...
...
tests/pipelines/spectrogram_diffusion/test_spectrogram_diffusion.py
View file @
1d2587bb
...
...
@@ -21,7 +21,7 @@ import torch
from
diffusers
import
DDPMScheduler
,
MidiProcessor
,
SpectrogramDiffusionPipeline
from
diffusers.pipelines.spectrogram_diffusion
import
SpectrogramContEncoder
,
SpectrogramNotesEncoder
,
T5FilmDecoder
from
diffusers.utils
import
require_torch_gpu
,
skip_mps
,
slow
,
torch_device
from
diffusers.utils
import
nightly
,
require_torch_gpu
,
skip_mps
,
torch_device
from
diffusers.utils.testing_utils
import
enable_full_determinism
,
require_note_seq
,
require_onnxruntime
from
..pipeline_params
import
TOKENS_TO_AUDIO_GENERATION_BATCH_PARAMS
,
TOKENS_TO_AUDIO_GENERATION_PARAMS
...
...
@@ -162,7 +162,7 @@ class SpectrogramDiffusionPipelineFastTests(PipelineTesterMixin, unittest.TestCa
return
super
().
test_progress_bar
()
@
slow
@
nightly
@
require_torch_gpu
@
require_onnxruntime
@
require_note_seq
...
...
tests/pipelines/versatile_diffusion/test_versatile_diffusion_image_variation.py
View file @
1d2587bb
...
...
@@ -19,7 +19,7 @@ import numpy as np
import
torch
from
diffusers
import
VersatileDiffusionImageVariationPipeline
from
diffusers.utils.testing_utils
import
load_image
,
require_torch_gpu
,
slow
,
torch_device
from
diffusers.utils.testing_utils
import
load_image
,
nightly
,
require_torch_gpu
,
torch_device
torch
.
backends
.
cuda
.
matmul
.
allow_tf32
=
False
...
...
@@ -29,7 +29,7 @@ class VersatileDiffusionImageVariationPipelineFastTests(unittest.TestCase):
pass
@
slow
@
nightly
@
require_torch_gpu
class
VersatileDiffusionImageVariationPipelineIntegrationTests
(
unittest
.
TestCase
):
def
test_inference_image_variations
(
self
):
...
...
tests/pipelines/vq_diffusion/test_vq_diffusion.py
View file @
1d2587bb
...
...
@@ -22,7 +22,7 @@ from transformers import CLIPTextConfig, CLIPTextModel, CLIPTokenizer
from
diffusers
import
Transformer2DModel
,
VQDiffusionPipeline
,
VQDiffusionScheduler
,
VQModel
from
diffusers.pipelines.vq_diffusion.pipeline_vq_diffusion
import
LearnedClassifierFreeSamplingEmbeddings
from
diffusers.utils
import
load_numpy
,
slow
,
torch_device
from
diffusers.utils
import
load_numpy
,
nightly
,
torch_device
from
diffusers.utils.testing_utils
import
require_torch_gpu
...
...
@@ -193,7 +193,7 @@ class VQDiffusionPipelineFastTests(unittest.TestCase):
assert
np
.
abs
(
image_from_tuple_slice
.
flatten
()
-
expected_slice
).
max
()
<
1e-2
@
slow
@
nightly
@
require_torch_gpu
class
VQDiffusionPipelineIntegrationTests
(
unittest
.
TestCase
):
def
tearDown
(
self
):
...
...
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