handle safetensors U8

9679520e · Michael Yang · c8ac4cc5 · 9679520e
Commit 9679520e authored Jul 24, 2025 by Michael Yang
Show whitespace changes
Inline Side-by-side

Showing with 3 additions and 0 deletions

convert/reader_safetensors.go convert/reader_safetensors.go +3 -0

No files found.
--- a/convert/reader_safetensors.go
+++ b/convert/reader_safetensors.go
@@ -150,6 +150,9 @@ func (st safetensor) WriteTo(w io.Writer) (int64, error) {
 		}

 		f32s = bfloat16.DecodeFloat32(u8s)
+	case "U8":
+		// U8 tensors do not support repacking or type conversion.
+		return io.CopyN(w, f, st.size)
 	default:
 		return 0, fmt.Errorf("unknown data type: %s", st.dtype)
 	}