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

Update the description

Browse source
This commit is contained in:
oobabooga 2023-01-07 16:54:49 -03:00
parent dd1bed2d8b
commit 45168e9e7a
1 changed files with 2 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

4
convert-to-torch.py
View file

@ -2,9 +2,9 @@
Converts a transformers model to .pt, which is faster to load. Converts a transformers model to .pt, which is faster to load.
Example: Example:
python convert.py models/opt-1.3b python convert-to-torch.py models/opt-1.3b
Output will be written to torch-dumps/name-of-the-model.pt The output will be written to torch-dumps/name-of-the-model.pt
''' '''
from transformers import AutoModelForCausalLM, T5ForConditionalGeneration from transformers import AutoModelForCausalLM, T5ForConditionalGeneration