Mirrors/text-generation-webui
1
0
Fork 0
mirror of https://github.com/oobabooga/text-generation-webui.git synced 2024-11-29 10:59:32 +01:00
Code Issues Packages Projects 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
Show Stats 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