thebrrt
/ 2safetensors.py
Created
December 27, 2023 11:15
— forked from epicfilemcnulty/2safetensors.py
convert pytorch weights to safetensors
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| import json | |
| import os | |
| import sys | |
| from collections import defaultdict | |
| from tqdm import tqdm | |
| import argparse | |
| import torch | |
| from safetensors.torch import load_file, save_file |
thebrrt
/ 2safetensors.py
Created
December 27, 2023 11:15
— forked from epicfilemcnulty/2safetensors.py
convert pytorch weights to safetensors
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| import json | |
| import os | |
| import sys | |
| from collections import defaultdict | |
| from tqdm import tqdm | |
| import argparse | |
| import torch | |
| from safetensors.torch import load_file, save_file |