Update README.md

This commit is contained in:
oobabooga 2023-02-12 00:47:03 -03:00 committed by GitHub
parent bf9dd8f8ee
commit 939e9d00a2
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -113,7 +113,7 @@ python download-model.py EleutherAI/gpt-j-6B --text-only
#### Converting to pytorch (optional) #### Converting to pytorch (optional)
The script `convert-to-torch.py` allows you to convert models to .pt format, which is sometimes 10x faster to load to the GPU: The script `convert-to-torch.py` allows you to convert models to .pt format, which can be a lot faster to load to the GPU:
python convert-to-torch.py models/model-name python convert-to-torch.py models/model-name