Skip to content

Commit 08cc36d

Browse files
authored
Fix MPS fast test warnings (#1744)
* unset level
1 parent 723e8f6 commit 08cc36d

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

tests/test_pipelines_common.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -233,11 +233,11 @@ def test_inference_batch_single_identical(self):
233233

234234
output = pipe(**inputs)
235235

236+
logger.setLevel(level=diffusers.logging.WARNING)
236237
if torch_device != "mps":
237238
# TODO(Pedro) - not sure why, but not at all reproducible at the moment it seems
238239
# make sure that batched and non-batched is identical
239240
assert np.abs(output_batch[0][0] - output[0][0]).max() < 1e-4
240-
logger.setLevel(level=diffusers.logging.WARNING)
241241

242242
def test_dict_tuple_outputs_equivalent(self):
243243
if torch_device == "mps" and self.pipeline_class in (

0 commit comments

Comments
 (0)