AI/text-generation-webui
1
0
Fork 0
mirror of https://github.com/oobabooga/text-generation-webui.git synced 2024-09-19 15:56:56 +00:00
Code Issues Packages Projects Releases Wiki Activity

Sort the imports

Browse Source
This commit is contained in:
oobabooga 2023-02-20 15:06:27 -03:00
parent 77846ceef3
commit 8123c5b01a
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
convert-to-safetensors.py
View File

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