Benjamin Gonzalez commited on
Commit
bc609bc
1 Parent(s): fb38431

add changes

Browse files
Files changed (3) hide show
  1. packages.txt +0 -1
  2. pre-requirements.txt +1 -0
  3. requirements.txt +0 -1
packages.txt DELETED
@@ -1 +0,0 @@
1
- nvidia-cuda-toolkit
 
 
pre-requirements.txt CHANGED
@@ -1,2 +1,3 @@
1
  --extra-index-url https://download.pytorch.org/whl/cu113
2
  torch==2.1.0
 
 
1
  --extra-index-url https://download.pytorch.org/whl/cu113
2
  torch==2.1.0
3
+ wheel
requirements.txt CHANGED
@@ -1,4 +1,3 @@
1
- flash-attn==2.3.6
2
  mlflow==2.6.0
3
  cloudpickle==2.2.1
4
  jsonpickle==3.0.1
 
 
1
  mlflow==2.6.0
2
  cloudpickle==2.2.1
3
  jsonpickle==3.0.1