Spaces:
Running
Running
Update diffusion_lens.py
Browse files- diffusion_lens.py +2 -2
diffusion_lens.py
CHANGED
@@ -8,8 +8,8 @@ model_dict = {
|
|
8 |
}
|
9 |
|
10 |
model_num_of_layers = {
|
11 |
-
'
|
12 |
-
'
|
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 |
|