Skip to content

Commit 5679067

Browse files
committed
make
1 parent 13a824e commit 5679067

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

tests/pipelines/test_pipelines.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -1818,16 +1818,16 @@ def test_wrong_model(self):
18181818
def test_wrong_model_scheduler_type(self):
18191819
scheduler = EulerDiscreteScheduler.from_pretrained("hf-internal-testing/tiny-flux-pipe", subfolder="scheduler")
18201820
with self.assertRaises(ValueError) as error_context:
1821-
_ = FluxPipeline.from_pretrained(
1822-
"hf-internal-testing/tiny-flux-pipe", scheduler=scheduler
1823-
)
1821+
_ = FluxPipeline.from_pretrained("hf-internal-testing/tiny-flux-pipe", scheduler=scheduler)
18241822

18251823
assert "Expected" in str(error_context.exception)
18261824
assert "scheduler" in str(error_context.exception)
18271825
assert "EulerDiscreteScheduler" in str(error_context.exception)
18281826

18291827
def test_wrong_model_scheduler_enum(self):
1830-
scheduler = FlowMatchEulerDiscreteScheduler.from_pretrained("hf-internal-testing/diffusers-stable-diffusion-tiny-all", subfolder="scheduler")
1828+
scheduler = FlowMatchEulerDiscreteScheduler.from_pretrained(
1829+
"hf-internal-testing/diffusers-stable-diffusion-tiny-all", subfolder="scheduler"
1830+
)
18311831
with self.assertRaises(ValueError) as error_context:
18321832
_ = StableDiffusionPipeline.from_pretrained(
18331833
"hf-internal-testing/diffusers-stable-diffusion-tiny-all", scheduler=scheduler

0 commit comments

Comments
 (0)