Skip to content

GitLab

Unverified Commit 9cc59ba0 authored by UmerHA's avatar UmerHA Committed by GitHub
Browse files

[Contributor Experience] Fix test collection on MPS (#6808) 

* Update testing_utils.py

* Update testing_utils.py
parent adcbe674
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 0 deletions
src/diffusers/utils/testing_utils.py
View file @ 9cc59ba0
...@@ -854,6 +854,8 @@ def _is_torch_fp64_available(device): ...@@ -854,6 +854,8 @@ def _is_torch_fp64_available(device):
import torch import torch
device = torch.device(device)
try: try:
x = torch.zeros((2, 2), dtype=torch.float64).to(device) x = torch.zeros((2, 2), dtype=torch.float64).to(device)
_ = torch.mul(x, x) _ = torch.mul(x, x)
......
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