Tonic commited on
Commit
28aebb4
1 Parent(s): 5363d0e

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +7 -4
requirements.txt CHANGED
@@ -1,5 +1,6 @@
1
- transformers
2
- torch
 
3
  einops
4
  accelerate
5
  tiktoken
@@ -10,9 +11,11 @@ bitsandbytes
10
  safetensors
11
  sentencepiece
12
  scipy
13
- torch==2.0.1
14
  torchaudio==2.0.2
15
  torchvision==0.15.2
16
  diffusers
17
  optimum
18
- auto-gptq
 
 
 
 
1
+ modelscope
2
+ transformers==4.32.0
3
+ torch==2.0.1
4
  einops
5
  accelerate
6
  tiktoken
 
11
  safetensors
12
  sentencepiece
13
  scipy
 
14
  torchaudio==2.0.2
15
  torchvision==0.15.2
16
  diffusers
17
  optimum
18
+ auto-gptq
19
+ pillow
20
+ tensorboard
21
+ matplotlib