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.
torch_device
1 parent 525d483 commit f34de43Copy full SHA for f34de43
1 file changed
tests/models/autoencoders/testing_utils.py
@@ -5,7 +5,8 @@
5
import torch
6
7
from diffusers.models.autoencoders.vae import DecoderOutput
8
-from diffusers.utils.torch_utils import torch_device
+
9
+from ...testing_utils import torch_device
10
11
12
class AutoencoderTesterMixin:
0 commit comments