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