Update diffusion_lens.py

diffusion_lens.py

@@ -8,8 +8,8 @@ model_dict = {
 }
 
 model_num_of_layers = {
-    'sd1': 12,
-    'sd2': 22,
+    'Stable Diffusion 1.4': 12,
+    'Stable Diffusion 2.1': 22,
 }