diff --git a/tests/test_helpers.py b/tests/test_helpers.py index a71c4756d8..334e9e745b 100644 --- a/tests/test_helpers.py +++ b/tests/test_helpers.py @@ -179,7 +179,7 @@ def test_scaling_set_to_zero(self, tokenizer): assert torch.allclose(logits_base_model, logits_lora_model) def test_diffusers_pipeline(self): - model_id = "hf-internal-testing/tiny-stable-diffusion-torch" + model_id = "hf-internal-testing/tiny-sd-pipe" pipeline = StableDiffusionPipeline.from_pretrained(model_id) text_encoder_kwargs = { diff --git a/tests/test_stablediffusion.py b/tests/test_stablediffusion.py index 53c06255eb..171d0d7c0f 100644 --- a/tests/test_stablediffusion.py +++ b/tests/test_stablediffusion.py @@ -36,7 +36,7 @@ from .testing_utils import temp_seed -PEFT_DIFFUSERS_SD_MODELS_TO_TEST = ["hf-internal-testing/tiny-stable-diffusion-torch"] +PEFT_DIFFUSERS_SD_MODELS_TO_TEST = ["hf-internal-testing/tiny-sd-pipe"] CONFIG_TESTING_KWARGS = ( { "text_encoder": { diff --git a/tests/test_tuners_utils.py b/tests/test_tuners_utils.py index 0896680c3b..8e56238125 100644 --- a/tests/test_tuners_utils.py +++ b/tests/test_tuners_utils.py @@ -331,7 +331,7 @@ def test_maybe_include_all_linear_layers_internals(self, initial_target_modules, assert new_config.target_modules == expected_target_modules def test_maybe_include_all_linear_layers_diffusion(self): - model_id = "hf-internal-testing/tiny-stable-diffusion-torch" + model_id = "hf-internal-testing/tiny-sd-pipe" model = StableDiffusionPipeline.from_pretrained(model_id) config = LoraConfig(base_model_name_or_path=model_id, target_modules="all-linear")