tokeron commited on
Commit
2a99ad2
1 Parent(s): 887521b

Update diffusion_lens.py

Browse files
Files changed (1) hide show
  1. diffusion_lens.py +2 -2
diffusion_lens.py CHANGED
@@ -2,8 +2,8 @@ from diffusers import StableDiffusionPipeline, EulerDiscreteScheduler
2
  import torch
3
 
4
  model_dict = {
5
- 'sd1': "CompVis/stable-diffusion-v1-4",
6
- 'sd2': "stabilityai/stable-diffusion-2-1",
7
  }
8
 
9
  model_num_of_layers = {
 
2
  import torch
3
 
4
  model_dict = {
5
+ 'Stable Diffusion 1.4': "CompVis/stable-diffusion-v1-4",
6
+ 'Stable Diffusion 2.1': "stabilityai/stable-diffusion-2-1",
7
  }
8
 
9
  model_num_of_layers = {