tokeron commited on
Commit
9035502
·
verified ·
1 Parent(s): d329d58

Update diffusion_lens.py

Browse files
Files changed (1) hide show
  1. diffusion_lens.py +2 -2
diffusion_lens.py CHANGED
@@ -8,8 +8,8 @@ model_dict = {
8
  }
9
 
10
  model_num_of_layers = {
11
- 'sd1': 12,
12
- 'sd2': 22,
13
  }
14
 
15
 
 
8
  }
9
 
10
  model_num_of_layers = {
11
+ 'Stable Diffusion 1.4': 12,
12
+ 'Stable Diffusion 2.1': 22,
13
  }
14
 
15