baubin commited on
Commit
81757a2
·
1 Parent(s): 2abc000

Update requirements.txt to allow flexible versions for sentencepiece, torch, and xformers; pin transformers version.

Browse files
Files changed (1) hide show
  1. requirements.txt +4 -5
requirements.txt CHANGED
@@ -7,11 +7,10 @@ numpy==1.26.4
7
  invisible_watermark
8
  optimum
9
  peft >= 0.6.0
10
- sentencepiece==0.2.0
11
  spaces
12
- --extra-index-url https://download.pytorch.org/whl/cu121
13
- torch==2.5.1
14
  torchaudio>=2.1.0
15
  torchvision>=0.16.0
16
- transformers>=4.34.0
17
- xformers>=0.0.22.post7
 
7
  invisible_watermark
8
  optimum
9
  peft >= 0.6.0
10
+ sentencepiece>=0.2.0
11
  spaces
12
+ torch
 
13
  torchaudio>=2.1.0
14
  torchvision>=0.16.0
15
+ transformers>=4.34.0,<4.49
16
+ xformers