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 a648319 commit cbdcdeaCopy full SHA for cbdcdea
packages/hub/src/lib/parse-safetensors-metadata.ts
@@ -42,7 +42,21 @@ class SafetensorParseError extends Error {}
42
type FileName = string;
43
44
export type TensorName = string;
45
-export type Dtype = "F64" | "F32" | "F16" | "BF16" | "I64" | "I32" | "I16" | "I8" | "U8" | "BOOL";
+export type Dtype =
46
+ | "F64"
47
+ | "F32"
48
+ | "F16"
49
+ | "F8_E4M3"
50
+ | "F8_E5M2"
51
+ | "BF16"
52
+ | "I64"
53
+ | "I32"
54
+ | "I16"
55
+ | "I8"
56
+ | "U16"
57
+ | "U8"
58
+ | "BOOL"
59
+ | "UNK"; /// when the total_parameters is stored directly in the header, we use this dummy dtype
60
61
export interface TensorInfo {
62
dtype: Dtype;
0 commit comments