BK-Lee commited on
Commit
b1ac5a9
1 Parent(s): c598547
Files changed (2) hide show
  1. README.md +1 -0
  2. requirements.txt +0 -1
README.md CHANGED
@@ -8,6 +8,7 @@ sdk_version: 4.31.5
8
  app_file: app.py
9
  pinned: false
10
  python_version: 3.11
 
11
  ---
12
 
13
  Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference
 
8
  app_file: app.py
9
  pinned: false
10
  python_version: 3.11
11
+ pytorch_version: 2.2.2
12
  ---
13
 
14
  Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference
requirements.txt CHANGED
@@ -1,4 +1,3 @@
1
- torch==2.2.2
2
  flash-attn --no-build-isolation
3
  transformers
4
  bitsandbytes
 
 
1
  flash-attn --no-build-isolation
2
  transformers
3
  bitsandbytes