Skip to content

Commit 43a338c

Browse files
committed
deprecate tests that use sd2base checkpoint.
1 parent d341f19 commit 43a338c

1 file changed

Lines changed: 4 additions & 0 deletions

File tree

src/maxdiffusion/tests/input_pipeline_interface_test.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -135,6 +135,7 @@ def test_make_dreambooth_train_iterator(self):
135135
cleanup(instance_class_local_dir)
136136
cleanup(class_class_local_dir)
137137

138+
@pytest.mark.skip("This test is deprecated and will be removed in a future version. Reason: stable diffusion 2 base is no longer in HuggingFace")
138139
def test_make_pokemon_hf_iterator(self):
139140
pyconfig.initialize(
140141
[
@@ -238,6 +239,7 @@ def test_make_pokemon_hf_iterator_sdxl(self):
238239
assert data["input_ids"].shape == (device_count, 2, 77)
239240
assert data["pixel_values"].shape == (device_count, 3, config.resolution, config.resolution)
240241

242+
@pytest.mark.skip("This test is deprecated and will be removed in a future version. Reason: stable diffusion 2 base is no longer in HuggingFace")
241243
def test_make_pokemon_tf_iterator_cache(self):
242244
pyconfig.initialize(
243245
[
@@ -300,6 +302,7 @@ def test_make_pokemon_tf_iterator_cache(self):
300302
config.resolution // vae_scale_factor,
301303
)
302304

305+
@pytest.mark.skip("This test is deprecated and will be removed in a future version. Reason: stable diffusion 2 base is no longer in HuggingFace")
303306
def test_make_pokemon_iterator_no_cache(self):
304307
pyconfig.initialize(
305308
[
@@ -550,6 +553,7 @@ def _parse_tfrecord_fn(example):
550553
8,
551554
)
552555

556+
@pytest.mark.skip("This test is deprecated and will be removed in a future version. Reason: stable diffusion 2 base is no longer in HuggingFace")
553557
def test_tfrecord(self):
554558
"""Validate latents match a deterministic output image"""
555559

0 commit comments

Comments
 (0)