Skip to content

GitLab

Switch branch/tag
Find file Normal viewHistoryPermalink
requirements.txt 201 Bytes
Newer Older
Suraj Patil's avatar
[Flax] ViT training example (#12300)  
Suraj Patil committed
1
2 
jax>=0.2.8
jaxlib>=0.1.59
Suraj Patil's avatar
bump flax version (#14343)  
Suraj Patil committed
3 
flax>=0.3.5
Suraj Patil's avatar
[Flax] ViT training example (#12300)  
Suraj Patil committed
4
5 
optax>=0.0.8
-f https://download.pytorch.org/whl/torch_stable.html
sshahrokhi's avatar
changing the requirements to a cpu torch version that works (#23483)  
sshahrokhi committed
6 
torch==1.11.0+cpu
Suraj Patil's avatar
[Flax] ViT training example (#12300)  
Suraj Patil committed
7 
-f https://download.pytorch.org/whl/torch_stable.html
sshahrokhi's avatar
changing the requirements to a cpu torch version that works (#23483)  
sshahrokhi committed
8 
torchvision==0.12.0+cpu