Fix test_components_function

hlky · hlky · commit 0a4864838ce4 · 2024-12-09T21:20:23.000Z
diff --git a/tests/pipelines/stable_diffusion_3/test_pipeline_stable_diffusion_3.py b/tests/pipelines/stable_diffusion_3/test_pipeline_stable_diffusion_3.py
@@ -103,6 +103,8 @@ def get_dummy_components(self):
             "tokenizer_3": tokenizer_3,
             "transformer": transformer,
             "vae": vae,
+            "image_encoder": None,
+            "feature_extractor": None,
         }
 
     def get_dummy_inputs(self, device, seed=0):
diff --git a/tests/pipelines/stable_diffusion_3/test_pipeline_stable_diffusion_3_img2img.py b/tests/pipelines/stable_diffusion_3/test_pipeline_stable_diffusion_3_img2img.py
@@ -105,6 +105,8 @@ def get_dummy_components(self):
             "tokenizer_3": tokenizer_3,
             "transformer": transformer,
             "vae": vae,
+            "image_encoder": None,
+            "feature_extractor": None,
         }
 
     def get_dummy_inputs(self, device, seed=0):
diff --git a/tests/pipelines/stable_diffusion_3/test_pipeline_stable_diffusion_3_inpaint.py b/tests/pipelines/stable_diffusion_3/test_pipeline_stable_diffusion_3_inpaint.py
@@ -106,6 +106,8 @@ def get_dummy_components(self):
             "tokenizer_3": tokenizer_3,
             "transformer": transformer,
             "vae": vae,
+            "image_encoder": None,
+            "feature_extractor": None,
         }
 
     def get_dummy_inputs(self, device, seed=0):

Original file line number	Diff line number	Diff line change
`@@ -103,6 +103,8 @@ def get_dummy_components(self):`
`103`	`103`	`"tokenizer_3": tokenizer_3,`
`104`	`104`	`"transformer": transformer,`
`105`	`105`	`"vae": vae,`
	`106`	`+ "image_encoder": None,`
	`107`	`+ "feature_extractor": None,`
`106`	`108`	`}`
`107`	`109`
`108`	`110`	`def get_dummy_inputs(self, device, seed=0):`
Original file line number	Diff line number	Diff line change
`@@ -105,6 +105,8 @@ def get_dummy_components(self):`
`105`	`105`	`"tokenizer_3": tokenizer_3,`
`106`	`106`	`"transformer": transformer,`
`107`	`107`	`"vae": vae,`
	`108`	`+ "image_encoder": None,`
	`109`	`+ "feature_extractor": None,`
`108`	`110`	`}`
`109`	`111`
`110`	`112`	`def get_dummy_inputs(self, device, seed=0):`
Original file line number	Diff line number	Diff line change
`@@ -106,6 +106,8 @@ def get_dummy_components(self):`
`106`	`106`	`"tokenizer_3": tokenizer_3,`
`107`	`107`	`"transformer": transformer,`
`108`	`108`	`"vae": vae,`
	`109`	`+ "image_encoder": None,`
	`110`	`+ "feature_extractor": None,`
`109`	`111`	`}`
`110`	`112`
`111`	`113`	`def get_dummy_inputs(self, device, seed=0):`