We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 49db233 commit d704a73Copy full SHA for d704a73
src/diffusers/utils/testing_utils.py
@@ -820,7 +820,9 @@ def _is_torch_fp16_available(device):
820
821
try:
822
x = torch.zeros((2, 2), dtype=torch.float16).to(device)
823
- _ = x @ x
+ _ = torch.mul(x, x)
824
+ return True
825
+
826
except Exception as e:
827
if device.type == "cuda":
828
raise ValueError(
@@ -838,7 +840,9 @@ def _is_torch_fp64_available(device):
838
840
839
841
842
x = torch.zeros((2, 2), dtype=torch.float64).to(device)
843
844
845
846
847
848
0 commit comments