LeroyWaa commited on
Commit
215dd59
·
1 Parent(s): cd1e40a

try pre requirements

Browse files
Files changed (2) hide show
  1. pre-requirements.txt +11 -0
  2. requirements.txt +0 -11
pre-requirements.txt ADDED
@@ -0,0 +1,11 @@
 
 
 
 
 
 
 
 
 
 
 
 
1
+ accelerate==0.30.1
2
+ torchvision==0.18.0
3
+ lpips==0.1.4
4
+ matplotlib
5
+ transformers==4.41.0
6
+ torch==2.3.0
7
+ lightning>2.3.1
8
+ tqdm==4.66.4
9
+ wandb
10
+ xformers==0.0.26.post1
11
+ git+https://github.com/janEbert/mup.git@fsdp-fix
requirements.txt CHANGED
@@ -1,14 +1,3 @@
1
- accelerate==0.30.1
2
- torchvision==0.18.0
3
- lpips==0.1.4
4
- matplotlib
5
- transformers==4.41.0
6
- torch==2.3.0
7
- lightning>2.3.1
8
- tqdm==4.66.4
9
- wandb
10
- xformers==0.0.26.post1
11
- git+https://github.com/janEbert/mup.git@fsdp-fix
12
 
13
  # For flash-attn
14
  wheel
 
 
 
 
 
 
 
 
 
 
 
 
1
 
2
  # For flash-attn
3
  wheel