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 5fb50f1 commit 76f79b3Copy full SHA for 76f79b3
src/diffusers/utils/remote_utils.py
@@ -15,7 +15,7 @@
15
16
17
if is_safetensors_available():
18
- from safetensors.torch import _tobytes
+ import safetensors
19
20
from ..image_processor import VaeImageProcessor
21
from ..video_processor import VideoProcessor
@@ -57,7 +57,7 @@ def remote_decode(
57
if height is not None and width is not None:
58
parameters["height"] = height
59
parameters["width"] = width
60
- tensor_data = _tobytes(tensor, "tensor")
+ tensor_data = safetensors.torch._tobytes(tensor, "tensor")
61
if input_tensor_type == "base64":
62
headers["Content-Type"] = "tensor/base64"
63
elif input_tensor_type == "binary":
0 commit comments