mmarch/text-generation-webui
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

Reorder the imports here too

Browse source
This commit is contained in:
oobabooga 2023-02-10 15:57:55 -03:00
parent 219366342b
commit 76d3d7ddb3
2 changed files with 6 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

9
convert-to-torch.py
View file

@ -6,11 +6,12 @@ python convert-to-torch.py models/opt-1.3b
The output will be written to torch-dumps/name-of-the-model.pt
'''
from transformers import AutoModelForCausalLM
import torch
from sys import argv
from pathlib import Path
from sys import argv
import torch
from transformers import AutoModelForCausalLM
path = Path(argv[1])
model_name = path.name