zhengjl/text-generation-webui
Public Access
1
0
Fork 0
Code Issues Pull Requests Releases 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