diff --git a/nodes.py b/nodes.py
index 1d1ae65..39d9605 100644
--- a/nodes.py
+++ b/nodes.py
@@ -7,7 +7,8 @@ from typing import cast, List, Literal, Optional, Union
 
 from diffusers.image_processor import VaeImageProcessor
 from diffusers.video_processor import VideoProcessor
-from safetensors.torch import _tobytes
+def _tobytes(tensor, name):
+    return tensor.contiguous().cpu().numpy().tobytes()
 
 DTYPE_MAP = {
     "float16": torch.float16,