Skip to content

Commit 76f79b3

Browse files
committed
test dependency
1 parent 5fb50f1 commit 76f79b3

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

src/diffusers/utils/remote_utils.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -15,7 +15,7 @@
1515

1616

1717
if is_safetensors_available():
18-
from safetensors.torch import _tobytes
18+
import safetensors
1919

2020
from ..image_processor import VaeImageProcessor
2121
from ..video_processor import VideoProcessor
@@ -57,7 +57,7 @@ def remote_decode(
5757
if height is not None and width is not None:
5858
parameters["height"] = height
5959
parameters["width"] = width
60-
tensor_data = _tobytes(tensor, "tensor")
60+
tensor_data = safetensors.torch._tobytes(tensor, "tensor")
6161
if input_tensor_type == "base64":
6262
headers["Content-Type"] = "tensor/base64"
6363
elif input_tensor_type == "binary":

0 commit comments

Comments
 (0)