oobabooga 2 лет назад
Родитель
Сommit
8123c5b01a
1 измененных файлов с 1 добавлено и 1 удалено
  1. 1 1
      convert-to-safetensors.py

+ 1 - 1
convert-to-safetensors.py

@@ -11,13 +11,13 @@ https://gist.github.com/81300/fe5b08bff1cba45296a829b9d6b0f303
 
 '''
 
+import argparse
 from pathlib import Path
 from sys import argv
 
 import torch
 from transformers import AutoModelForCausalLM
 from transformers import AutoTokenizer
-import argparse
  
 parser = argparse.ArgumentParser(formatter_class=lambda prog: argparse.HelpFormatter(prog,max_help_position=54))
 parser.add_argument('MODEL', type=str, default=None, nargs='?', help="Path to the input model.")