Update diffusion_lens.py

diffusion_lens.py

@@ -13,7 +13,7 @@ pipeline.to(device)
 def get_images(prompt, skip_layers):
     print('inside get images')
     print(f'skipping {skip_layers}')
-    pipeline_output = pipeline(prompt, skip_layers=skip_layers, num_images_per_prompt=1, return_tensors=False)
+    pipeline_output = pipeline(prompt, clip_skip=skip_layers, num_images_per_prompt=1, return_tensors=False)
     print('after pipeline')
     images = pipeline_output.images
     print('got images')