Skip to content

GitLab

Unverified Commit 08cc36dd authored by Anton Lozhkov's avatar Anton Lozhkov Committed by GitHub
Browse files

Fix MPS fast test warnings (#1744) 

* unset level
parent 723e8f6b
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
tests/test_pipelines_common.py
View file @ 08cc36dd
...@@ -233,11 +233,11 @@ class PipelineTesterMixin: ...@@ -233,11 +233,11 @@ class PipelineTesterMixin:
output = pipe(**inputs) output = pipe(**inputs)
logger.setLevel(level=diffusers.logging.WARNING)
if torch_device != "mps": if torch_device != "mps":
# TODO(Pedro) - not sure why, but not at all reproducible at the moment it seems # TODO(Pedro) - not sure why, but not at all reproducible at the moment it seems
# make sure that batched and non-batched is identical # make sure that batched and non-batched is identical
assert np.abs(output_batch[0][0] - output[0][0]).max() < 1e-4 assert np.abs(output_batch[0][0] - output[0][0]).max() < 1e-4
logger.setLevel(level=diffusers.logging.WARNING)
def test_dict_tuple_outputs_equivalent(self): def test_dict_tuple_outputs_equivalent(self):
if torch_device == "mps" and self.pipeline_class in ( if torch_device == "mps" and self.pipeline_class in (
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment