Skip to content

Commit

Permalink
minor fixes
Browse files Browse the repository at this point in the history
  • Loading branch information
imangohari1 committed Feb 5, 2025
1 parent 5e55455 commit 2c4e06e
Showing 1 changed file with 1 addition and 2 deletions.
3 changes: 1 addition & 2 deletions tests/test_diffusers.py
Original file line number Diff line number Diff line change
Expand Up @@ -205,7 +205,7 @@ def check_8xhpu(test_case):

def legacy(test_case):
"""
Decorator skipping a test if it's old
Decorator used to skip tests for legacy models
"""
skip = os.environ.get("RUN_DIFFUSERS_LEGACY", "0") != "1"
return pytest.mark.skipif(skip, reason="This test is for old/legacy mmodel. Skipped starting 1.16.0")(test_case)
Expand Down Expand Up @@ -6025,7 +6025,6 @@ def test_flux_prompt_embeds(self):
@slow
@check_gated_model_access("black-forest-labs/FLUX.1-dev")
@pytest.mark.skipif(not IS_GAUDI2, reason="does not fit into Gaudi1 memory")
@legacy
def test_flux_img2img_inference(self):
repo_id = "black-forest-labs/FLUX.1-dev"
image_path = "https://huggingface.co/datasets/huggingface/documentation-images/resolve/main/diffusers/cat.png"
Expand Down

0 comments on commit 2c4e06e

Please sign in to comment.