Skip to content

Commit f34de43

Browse files
authored
[CI] Fix torch_device import in AutoencoderTesterMixin (#13852)
update
1 parent 525d483 commit f34de43

1 file changed

Lines changed: 2 additions & 1 deletion

File tree

tests/models/autoencoders/testing_utils.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -5,7 +5,8 @@
55
import torch
66

77
from diffusers.models.autoencoders.vae import DecoderOutput
8-
from diffusers.utils.torch_utils import torch_device
8+
9+
from ...testing_utils import torch_device
910

1011

1112
class AutoencoderTesterMixin:

0 commit comments

Comments
 (0)