Relax torch version to allow 2.1.x (useful for colab)
This commit is contained in:
@@ -24,6 +24,6 @@ requests
|
|||||||
safetensors
|
safetensors
|
||||||
scipy
|
scipy
|
||||||
tensorboard
|
tensorboard
|
||||||
torch==2.1.2 # For users without GPU
|
torch>=2.1,<2.2 # For users without GPU or colab
|
||||||
transformers
|
transformers
|
||||||
umap-learn
|
umap-learn
|
||||||
Reference in New Issue
Block a user