Further refactor

2023-02-23 13:28:30 -03:00
parent ce7feb3641
commit 1dacd34165
8 changed files with 185 additions and 173 deletions
--- a/convert-to-safetensors.py
+++ b/convert-to-safetensors.py
@@ -12,7 +12,6 @@ https://gist.github.com/81300/fe5b08bff1cba45296a829b9d6b0f303
 '''
 import argparse
 from pathlib import Path
-from sys import argv

 import torch
 from transformers import AutoModelForCausalLM