Spaces:
Sleeping
Sleeping
Update diffusion_lens.py
Browse files- diffusion_lens.py +2 -2
diffusion_lens.py
CHANGED
|
@@ -9,8 +9,8 @@ model_dict = {
|
|
| 9 |
}
|
| 10 |
|
| 11 |
model_num_of_layers = {
|
| 12 |
-
'Stable Diffusion 1.4':
|
| 13 |
-
'Stable Diffusion 2.1':
|
| 14 |
}
|
| 15 |
|
| 16 |
|
|
|
|
| 9 |
}
|
| 10 |
|
| 11 |
model_num_of_layers = {
|
| 12 |
+
'Stable Diffusion 1.4': 13,
|
| 13 |
+
'Stable Diffusion 2.1': 23,
|
| 14 |
}
|
| 15 |
|
| 16 |
|