Update PyTorch dependencies in requirements.txt
Removed comments about CUDA version and updated PyTorch dependencies.
This commit is contained in:
1 parent
0d7db1cd4e
commit
271c81b734
1 file changed
+15
@@ -0,0 +1,15 @@
|
|||||||
|
fastapi
|
||||||
|
uvicorn[standard]
|
||||||
|
pydantic
|
||||||
|
python-multipart
|
||||||
|
|
||||||
|
transformers==4.37.2
|
||||||
|
tokenizers
|
||||||
|
sentencepiece
|
||||||
|
|
||||||
|
torch==2.2.1+cu121
|
||||||
|
torchvision==0.17.1+cu121
|
||||||
|
torchaudio==2.2.1+cu121
|
||||||
|
--extra-index-url https://download.pytorch.org/whl/cu121
|
||||||
|
|
||||||
|
numpy
|
||||||
Reference in new issue
Block a user