FagerholmEmil commited on
Commit
cdc4fc4
·
1 Parent(s): 5bd82bc

Update project dependencies with expanded and more specific requirements

Browse files
Files changed (1) hide show
  1. requirements.txt +19 -4
requirements.txt CHANGED
@@ -1,4 +1,19 @@
1
- gradio==4.19.2
2
- transformer_lens==1.13.0
3
- torch==2.2.0
4
- pytest>=7.0.0
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ accelerate>=0.23.0
2
+ beartype>=0.14.1
3
+ better-abc>=0.0.3
4
+ datasets>=2.7.1
5
+ einops>=0.6.0
6
+ fancy-einsum>=0.0.3
7
+ jaxtyping>=0.2.11
8
+ numpy>=1.24 # Assuming Python 3.9+
9
+ pandas>=1.1.5
10
+ rich>=12.6.0
11
+ sentencepiece
12
+ torch>=1.10
13
+ tqdm>=4.64.1
14
+ transformers>=4.41,<4.42 # Assuming Python 3.9+
15
+ typing-extensions
16
+ wandb>=0.13.5
17
+ typeguard>=4.2
18
+ pytest>=7.2.0 # From dev dependencies, needed by transformer_lens
19
+ gradio==4.19.2 # Keeping your existing Gradio requirement