Skip to content

Commit

Permalink
Added safetensors files.
Browse files Browse the repository at this point in the history
  • Loading branch information
KOLANICH committed Jan 8, 2023
1 parent 8a328c3 commit 3de6b70
Show file tree
Hide file tree
Showing 94 changed files with 173 additions and 0 deletions.
Binary file added serialization/safetensors/B.safetensors
Binary file not shown.
2 changes: 2 additions & 0 deletions serialization/safetensors/B.safetensors.license
Original file line number Diff line number Diff line change
@@ -0,0 +1,2 @@
SPDX-FileCopyrightText: Uncopyrightable
SPDX-License-Identifier: Unlicense
Binary file added serialization/safetensors/B2x2x2.safetensors
Binary file not shown.
2 changes: 2 additions & 0 deletions serialization/safetensors/B2x2x2.safetensors.license
Original file line number Diff line number Diff line change
@@ -0,0 +1,2 @@
SPDX-FileCopyrightText: Uncopyrightable
SPDX-License-Identifier: Unlicense
Binary file added serialization/safetensors/B2x4.safetensors
Binary file not shown.
2 changes: 2 additions & 0 deletions serialization/safetensors/B2x4.safetensors.license
Original file line number Diff line number Diff line change
@@ -0,0 +1,2 @@
SPDX-FileCopyrightText: Uncopyrightable
SPDX-License-Identifier: Unlicense
Binary file added serialization/safetensors/B4x2.safetensors
Binary file not shown.
2 changes: 2 additions & 0 deletions serialization/safetensors/B4x2.safetensors.license
Original file line number Diff line number Diff line change
@@ -0,0 +1,2 @@
SPDX-FileCopyrightText: Uncopyrightable
SPDX-License-Identifier: Unlicense
Binary file added serialization/safetensors/B8.safetensors
Binary file not shown.
2 changes: 2 additions & 0 deletions serialization/safetensors/B8.safetensors.license
Original file line number Diff line number Diff line change
@@ -0,0 +1,2 @@
SPDX-FileCopyrightText: Uncopyrightable
SPDX-License-Identifier: Unlicense
Binary file added serialization/safetensors/H.safetensors
Binary file not shown.
2 changes: 2 additions & 0 deletions serialization/safetensors/H.safetensors.license
Original file line number Diff line number Diff line change
@@ -0,0 +1,2 @@
SPDX-FileCopyrightText: Uncopyrightable
SPDX-License-Identifier: Unlicense
Binary file added serialization/safetensors/H2x2x2.safetensors
Binary file not shown.
2 changes: 2 additions & 0 deletions serialization/safetensors/H2x2x2.safetensors.license
Original file line number Diff line number Diff line change
@@ -0,0 +1,2 @@
SPDX-FileCopyrightText: Uncopyrightable
SPDX-License-Identifier: Unlicense
Binary file added serialization/safetensors/H2x4.safetensors
Binary file not shown.
2 changes: 2 additions & 0 deletions serialization/safetensors/H2x4.safetensors.license
Original file line number Diff line number Diff line change
@@ -0,0 +1,2 @@
SPDX-FileCopyrightText: Uncopyrightable
SPDX-License-Identifier: Unlicense
Binary file added serialization/safetensors/H4x2.safetensors
Binary file not shown.
2 changes: 2 additions & 0 deletions serialization/safetensors/H4x2.safetensors.license
Original file line number Diff line number Diff line change
@@ -0,0 +1,2 @@
SPDX-FileCopyrightText: Uncopyrightable
SPDX-License-Identifier: Unlicense
Binary file added serialization/safetensors/H8.safetensors
Binary file not shown.
2 changes: 2 additions & 0 deletions serialization/safetensors/H8.safetensors.license
Original file line number Diff line number Diff line change
@@ -0,0 +1,2 @@
SPDX-FileCopyrightText: Uncopyrightable
SPDX-License-Identifier: Unlicense
Binary file added serialization/safetensors/I.safetensors
Binary file not shown.
2 changes: 2 additions & 0 deletions serialization/safetensors/I.safetensors.license
Original file line number Diff line number Diff line change
@@ -0,0 +1,2 @@
SPDX-FileCopyrightText: Uncopyrightable
SPDX-License-Identifier: Unlicense
Binary file added serialization/safetensors/I2x2x2.safetensors
Binary file not shown.
2 changes: 2 additions & 0 deletions serialization/safetensors/I2x2x2.safetensors.license
Original file line number Diff line number Diff line change
@@ -0,0 +1,2 @@
SPDX-FileCopyrightText: Uncopyrightable
SPDX-License-Identifier: Unlicense
Binary file added serialization/safetensors/I2x4.safetensors
Binary file not shown.
2 changes: 2 additions & 0 deletions serialization/safetensors/I2x4.safetensors.license
Original file line number Diff line number Diff line change
@@ -0,0 +1,2 @@
SPDX-FileCopyrightText: Uncopyrightable
SPDX-License-Identifier: Unlicense
Binary file added serialization/safetensors/I4x2.safetensors
Binary file not shown.
2 changes: 2 additions & 0 deletions serialization/safetensors/I4x2.safetensors.license
Original file line number Diff line number Diff line change
@@ -0,0 +1,2 @@
SPDX-FileCopyrightText: Uncopyrightable
SPDX-License-Identifier: Unlicense
Binary file added serialization/safetensors/I8.safetensors
Binary file not shown.
2 changes: 2 additions & 0 deletions serialization/safetensors/I8.safetensors.license
Original file line number Diff line number Diff line change
@@ -0,0 +1,2 @@
SPDX-FileCopyrightText: Uncopyrightable
SPDX-License-Identifier: Unlicense
8 changes: 8 additions & 0 deletions serialization/safetensors/ReadMe.md
Original file line number Diff line number Diff line change
@@ -0,0 +1,8 @@
<!--
SPDX-FileCopyrightText: 2022 KOLANICH
SPDX-License-Identifier: Unlicense
-->

# HuggingFace🤗 `safetensors` format

These are the samples of [`safetensors` format](https://github.com/huggingface/safetensors) used for serialization of models for HuggingFace Hub.
Binary file added serialization/safetensors/b.safetensors
Binary file not shown.
2 changes: 2 additions & 0 deletions serialization/safetensors/b.safetensors.license
Original file line number Diff line number Diff line change
@@ -0,0 +1,2 @@
SPDX-FileCopyrightText: Uncopyrightable
SPDX-License-Identifier: Unlicense
Binary file added serialization/safetensors/b2x2x2.safetensors
Binary file not shown.
2 changes: 2 additions & 0 deletions serialization/safetensors/b2x2x2.safetensors.license
Original file line number Diff line number Diff line change
@@ -0,0 +1,2 @@
SPDX-FileCopyrightText: Uncopyrightable
SPDX-License-Identifier: Unlicense
Binary file added serialization/safetensors/b2x4.safetensors
Binary file not shown.
2 changes: 2 additions & 0 deletions serialization/safetensors/b2x4.safetensors.license
Original file line number Diff line number Diff line change
@@ -0,0 +1,2 @@
SPDX-FileCopyrightText: Uncopyrightable
SPDX-License-Identifier: Unlicense
Binary file added serialization/safetensors/b4x2.safetensors
Binary file not shown.
2 changes: 2 additions & 0 deletions serialization/safetensors/b4x2.safetensors.license
Original file line number Diff line number Diff line change
@@ -0,0 +1,2 @@
SPDX-FileCopyrightText: Uncopyrightable
SPDX-License-Identifier: Unlicense
Binary file added serialization/safetensors/b8.safetensors
Binary file not shown.
2 changes: 2 additions & 0 deletions serialization/safetensors/b8.safetensors.license
Original file line number Diff line number Diff line change
@@ -0,0 +1,2 @@
SPDX-FileCopyrightText: Uncopyrightable
SPDX-License-Identifier: Unlicense
Binary file added serialization/safetensors/d.safetensors
Binary file not shown.
2 changes: 2 additions & 0 deletions serialization/safetensors/d.safetensors.license
Original file line number Diff line number Diff line change
@@ -0,0 +1,2 @@
SPDX-FileCopyrightText: Uncopyrightable
SPDX-License-Identifier: Unlicense
Binary file added serialization/safetensors/d2x2x2.safetensors
Binary file not shown.
2 changes: 2 additions & 0 deletions serialization/safetensors/d2x2x2.safetensors.license
Original file line number Diff line number Diff line change
@@ -0,0 +1,2 @@
SPDX-FileCopyrightText: Uncopyrightable
SPDX-License-Identifier: Unlicense
Binary file added serialization/safetensors/d2x4.safetensors
Binary file not shown.
2 changes: 2 additions & 0 deletions serialization/safetensors/d2x4.safetensors.license
Original file line number Diff line number Diff line change
@@ -0,0 +1,2 @@
SPDX-FileCopyrightText: Uncopyrightable
SPDX-License-Identifier: Unlicense
Binary file added serialization/safetensors/d4x2.safetensors
Binary file not shown.
2 changes: 2 additions & 0 deletions serialization/safetensors/d4x2.safetensors.license
Original file line number Diff line number Diff line change
@@ -0,0 +1,2 @@
SPDX-FileCopyrightText: Uncopyrightable
SPDX-License-Identifier: Unlicense
Binary file added serialization/safetensors/d8.safetensors
Binary file not shown.
2 changes: 2 additions & 0 deletions serialization/safetensors/d8.safetensors.license
Original file line number Diff line number Diff line change
@@ -0,0 +1,2 @@
SPDX-FileCopyrightText: Uncopyrightable
SPDX-License-Identifier: Unlicense
Binary file added serialization/safetensors/e.safetensors
Binary file not shown.
2 changes: 2 additions & 0 deletions serialization/safetensors/e.safetensors.license
Original file line number Diff line number Diff line change
@@ -0,0 +1,2 @@
SPDX-FileCopyrightText: Uncopyrightable
SPDX-License-Identifier: Unlicense
Binary file added serialization/safetensors/e2x2x2.safetensors
Binary file not shown.
2 changes: 2 additions & 0 deletions serialization/safetensors/e2x2x2.safetensors.license
Original file line number Diff line number Diff line change
@@ -0,0 +1,2 @@
SPDX-FileCopyrightText: Uncopyrightable
SPDX-License-Identifier: Unlicense
Binary file added serialization/safetensors/e2x4.safetensors
Binary file not shown.
2 changes: 2 additions & 0 deletions serialization/safetensors/e2x4.safetensors.license
Original file line number Diff line number Diff line change
@@ -0,0 +1,2 @@
SPDX-FileCopyrightText: Uncopyrightable
SPDX-License-Identifier: Unlicense
Binary file added serialization/safetensors/e4x2.safetensors
Binary file not shown.
2 changes: 2 additions & 0 deletions serialization/safetensors/e4x2.safetensors.license
Original file line number Diff line number Diff line change
@@ -0,0 +1,2 @@
SPDX-FileCopyrightText: Uncopyrightable
SPDX-License-Identifier: Unlicense
Binary file added serialization/safetensors/e8.safetensors
Binary file not shown.
2 changes: 2 additions & 0 deletions serialization/safetensors/e8.safetensors.license
Original file line number Diff line number Diff line change
@@ -0,0 +1,2 @@
SPDX-FileCopyrightText: Uncopyrightable
SPDX-License-Identifier: Unlicense
Binary file added serialization/safetensors/f.safetensors
Binary file not shown.
2 changes: 2 additions & 0 deletions serialization/safetensors/f.safetensors.license
Original file line number Diff line number Diff line change
@@ -0,0 +1,2 @@
SPDX-FileCopyrightText: Uncopyrightable
SPDX-License-Identifier: Unlicense
Binary file added serialization/safetensors/f2x2x2.safetensors
Binary file not shown.
2 changes: 2 additions & 0 deletions serialization/safetensors/f2x2x2.safetensors.license
Original file line number Diff line number Diff line change
@@ -0,0 +1,2 @@
SPDX-FileCopyrightText: Uncopyrightable
SPDX-License-Identifier: Unlicense
Binary file added serialization/safetensors/f2x4.safetensors
Binary file not shown.
2 changes: 2 additions & 0 deletions serialization/safetensors/f2x4.safetensors.license
Original file line number Diff line number Diff line change
@@ -0,0 +1,2 @@
SPDX-FileCopyrightText: Uncopyrightable
SPDX-License-Identifier: Unlicense
Binary file added serialization/safetensors/f4x2.safetensors
Binary file not shown.
2 changes: 2 additions & 0 deletions serialization/safetensors/f4x2.safetensors.license
Original file line number Diff line number Diff line change
@@ -0,0 +1,2 @@
SPDX-FileCopyrightText: Uncopyrightable
SPDX-License-Identifier: Unlicense
Binary file added serialization/safetensors/f8.safetensors
Binary file not shown.
2 changes: 2 additions & 0 deletions serialization/safetensors/f8.safetensors.license
Original file line number Diff line number Diff line change
@@ -0,0 +1,2 @@
SPDX-FileCopyrightText: Uncopyrightable
SPDX-License-Identifier: Unlicense
73 changes: 73 additions & 0 deletions serialization/safetensors/generate.py
Original file line number Diff line number Diff line change
@@ -0,0 +1,73 @@
#!/usr/bin/env python3

from pathlib import Path

import numpy as np
import safetensors.numpy

import safetensors


__author__ = "KOLANICH"
__copyright__ = "Public domain"
__license__ = "Unlicense"

__reuse__ = """SPDX-FileCopyrightText: Uncopyrightable
SPDX-License-Identifier: Unlicense
"""

thisDir = Path(".").absolute()


ambigiousTypes = set("LPQlpq") # these letters result into the same arrays


def makeTypeCodes():
typeCodes = "".join(
sorted(set("".join(np.typecodes[el] for el in ("Complex", "AllInteger", "AllFloat"))))
)
typeCodes = "".join(set(typeCodes) - ambigiousTypes)
return typeCodes


def makeNumpyDict(typeCodes) -> dict:
npDict = {}
shapes = (
(8,),
(2, 4),
(4, 2),
(2, 2, 2),
)
for t in typeCodes:
a = np.array(range(-3, 5), dtype="<" + t)
try:
testDic = {t + "x".join(str(el) for el in shape): a.reshape(shape) for shape in shapes}
safetensors.numpy.save(testDic) # testing if the type safetensors-serializeable
except BaseException as ex: # pylint:disable=broad-except
print(t, a.dtype, ex)
else:
npDict.update(testDic)
return npDict


EXT = ".safetensors"


def dumpFile(name: str, data: bytes):
fn = name + EXT
(thisDir / fn).write_bytes(data)
(thisDir / (fn + ".license")).write_text(__reuse__)


def main():
typeCodes = makeTypeCodes()
npDict = makeNumpyDict(typeCodes)
metaDict = {"test": "a"}
dumpFile("overall", safetensors.numpy.save(npDict, metaDict))

for k, v in npDict.items():
dumpFile(k, safetensors.numpy.save({k: v}, metaDict))


if __name__ == "__main__":
main()
Binary file added serialization/safetensors/h.safetensors
Binary file not shown.
2 changes: 2 additions & 0 deletions serialization/safetensors/h.safetensors.license
Original file line number Diff line number Diff line change
@@ -0,0 +1,2 @@
SPDX-FileCopyrightText: Uncopyrightable
SPDX-License-Identifier: Unlicense
Binary file added serialization/safetensors/h2x2x2.safetensors
Binary file not shown.
2 changes: 2 additions & 0 deletions serialization/safetensors/h2x2x2.safetensors.license
Original file line number Diff line number Diff line change
@@ -0,0 +1,2 @@
SPDX-FileCopyrightText: Uncopyrightable
SPDX-License-Identifier: Unlicense
Binary file added serialization/safetensors/h2x4.safetensors
Binary file not shown.
2 changes: 2 additions & 0 deletions serialization/safetensors/h2x4.safetensors.license
Original file line number Diff line number Diff line change
@@ -0,0 +1,2 @@
SPDX-FileCopyrightText: Uncopyrightable
SPDX-License-Identifier: Unlicense
Binary file added serialization/safetensors/h4x2.safetensors
Binary file not shown.
2 changes: 2 additions & 0 deletions serialization/safetensors/h4x2.safetensors.license
Original file line number Diff line number Diff line change
@@ -0,0 +1,2 @@
SPDX-FileCopyrightText: Uncopyrightable
SPDX-License-Identifier: Unlicense
Binary file added serialization/safetensors/h8.safetensors
Binary file not shown.
2 changes: 2 additions & 0 deletions serialization/safetensors/h8.safetensors.license
Original file line number Diff line number Diff line change
@@ -0,0 +1,2 @@
SPDX-FileCopyrightText: Uncopyrightable
SPDX-License-Identifier: Unlicense
Binary file added serialization/safetensors/i.safetensors
Binary file not shown.
2 changes: 2 additions & 0 deletions serialization/safetensors/i.safetensors.license
Original file line number Diff line number Diff line change
@@ -0,0 +1,2 @@
SPDX-FileCopyrightText: Uncopyrightable
SPDX-License-Identifier: Unlicense
Binary file added serialization/safetensors/i2x2x2.safetensors
Binary file not shown.
2 changes: 2 additions & 0 deletions serialization/safetensors/i2x2x2.safetensors.license
Original file line number Diff line number Diff line change
@@ -0,0 +1,2 @@
SPDX-FileCopyrightText: Uncopyrightable
SPDX-License-Identifier: Unlicense
Binary file added serialization/safetensors/i2x4.safetensors
Binary file not shown.
2 changes: 2 additions & 0 deletions serialization/safetensors/i2x4.safetensors.license
Original file line number Diff line number Diff line change
@@ -0,0 +1,2 @@
SPDX-FileCopyrightText: Uncopyrightable
SPDX-License-Identifier: Unlicense
Binary file added serialization/safetensors/i4x2.safetensors
Binary file not shown.
2 changes: 2 additions & 0 deletions serialization/safetensors/i4x2.safetensors.license
Original file line number Diff line number Diff line change
@@ -0,0 +1,2 @@
SPDX-FileCopyrightText: Uncopyrightable
SPDX-License-Identifier: Unlicense
Binary file added serialization/safetensors/i8.safetensors
Binary file not shown.
2 changes: 2 additions & 0 deletions serialization/safetensors/i8.safetensors.license
Original file line number Diff line number Diff line change
@@ -0,0 +1,2 @@
SPDX-FileCopyrightText: Uncopyrightable
SPDX-License-Identifier: Unlicense
Binary file added serialization/safetensors/overall.safetensors
Binary file not shown.
2 changes: 2 additions & 0 deletions serialization/safetensors/overall.safetensors.license
Original file line number Diff line number Diff line change
@@ -0,0 +1,2 @@
SPDX-FileCopyrightText: Uncopyrightable
SPDX-License-Identifier: Unlicense

0 comments on commit 3de6b70

Please sign in to comment.