WoodenMaiden's picture
fix(deps): add pytorch
daa9219 unverified
raw
history blame
No virus
413 Bytes
[[source]]
url = "https://pypi.org/simple"
verify_ssl = true
name = "pypi"
[[source]]
name = "pytorch"
url = "https://download.pytorch.org/whl/"
verify_ssl = true
[packages]
transformers = "*"
numpy = "*"
torch = {index = "pytorch",version = "*"}
torchvision = {index ="pytorch", version= "*"}
matplotlib = "*"
torchaudio = {version = "*", index = "pytorch"}
[dev-packages]
[requires]
python_version = "3.11"