File tree Expand file tree Collapse file tree 1 file changed +4
-4
lines changed Expand file tree Collapse file tree 1 file changed +4
-4
lines changed Original file line number Diff line number Diff 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
You can’t perform that action at this time.
0 commit comments