tokeron commited on
Commit
1bfc07b
1 Parent(s): 7eec65f

Update diffusion_lens.py

Browse files
Files changed (1) hide show
  1. diffusion_lens.py +1 -1
diffusion_lens.py CHANGED
@@ -33,7 +33,7 @@ def get_images(prompt, skip_layers, model, seed):
33
  pipeline.scheduler = EulerDiscreteScheduler.from_config(pipeline.scheduler.config)
34
  print('inside get images')
35
  layer = model_num_of_layers[model] - skip_layers
36
- gr.Info(f"Generating image from {layer}'th layer")
37
  print(f'skipping {skip_layers}')
38
  pipeline_output = pipeline(prompt, clip_skip=skip_layers, num_images_per_prompt=1, return_tensors=False, seed=seed)
39
  print('after pipeline')
 
33
  pipeline.scheduler = EulerDiscreteScheduler.from_config(pipeline.scheduler.config)
34
  print('inside get images')
35
  layer = model_num_of_layers[model] - skip_layers
36
+ gr.Info(f"Generating an image from {layer}'th layer")
37
  print(f'skipping {skip_layers}')
38
  pipeline_output = pipeline(prompt, clip_skip=skip_layers, num_images_per_prompt=1, return_tensors=False, seed=seed)
39
  print('after pipeline')