@@ -111,31 +111,31 @@ def get_dummy_inputs(self, device, seed=0):
111111            "generator" : generator ,
112112            "image" : image ,
113113            "mask_image" : mask_image ,
114-             "strength" : 0.9  ,
115-             "num_inference_steps" : 4 ,
116-             "guidance_scale" : 5 .0
114+             "strength" : 0.8  ,
115+             "num_inference_steps" : 2 ,
116+             "guidance_scale" : 30 .0
117117            "height" : 8 ,
118118            "width" : 8 ,
119119            "max_sequence_length" : 48 ,
120120            "output_type" : "np" ,
121121        }
122122        return  inputs 
123123
124-     def  test_flux_different_prompts (self ):
125-         pipe  =  self .pipeline_class (** self .get_dummy_components ()).to (torch_device )
124+     #  def test_flux_different_prompts(self):
125+     #      pipe = self.pipeline_class(**self.get_dummy_components()).to(torch_device)
126126
127-         inputs  =  self .get_dummy_inputs (torch_device )
128-         output_same_prompt  =  pipe (** inputs ).images [0 ]
127+     #      inputs = self.get_dummy_inputs(torch_device)
128+     #      output_same_prompt = pipe(**inputs).images[0]
129129
130-         inputs  =  self .get_dummy_inputs (torch_device )
131-         inputs ["prompt_2" ] =  "a different prompt" 
132-         output_different_prompts  =  pipe (** inputs ).images [0 ]
130+     #      inputs = self.get_dummy_inputs(torch_device)
131+     #      inputs["prompt_2"] = "a different prompt"
132+     #      output_different_prompts = pipe(**inputs).images[0]
133133
134-         max_diff  =  np .abs (output_same_prompt  -  output_different_prompts ).max ()
134+     #      max_diff = np.abs(output_same_prompt - output_different_prompts).max()
135135
136-         # Outputs should be different here 
137-         # For some reasons, they don't show large differences 
138-         assert  max_diff  >  1e-6 
136+     #      # Outputs should be different here
137+     #      # For some reasons, they don't show large differences
138+     #      assert max_diff > 1e-6
139139
140140    def  test_flux_prompt_embeds (self ):
141141        pipe  =  self .pipeline_class (** self .get_dummy_components ()).to (torch_device )
0 commit comments