clementchadebec commited on
Commit
e23ae33
1 Parent(s): f892d4a

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +5 -1
requirements.txt CHANGED
@@ -1,7 +1,11 @@
1
  accelerate
2
  diffusers
3
  invisible_watermark
4
- torch
 
5
  peft >= 0.6.0
 
 
 
6
  transformers >= 4.34.0
7
  xformers
 
1
  accelerate
2
  diffusers
3
  invisible_watermark
4
+ --extra-index-url https://download.pytorch.org/whl/cu118
5
+ torch==2.0.1
6
  peft >= 0.6.0
7
+ sentencepiece==0.2.0
8
+ optimum
9
+ beautifulsoup4
10
  transformers >= 4.34.0
11
  xformers