diff --git a/convert-to-safetensors.py b/convert-to-safetensors.py index 177c67d0..60770843 100644 --- a/convert-to-safetensors.py +++ b/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.")