tokeron commited on
Commit
69b9e7f
1 Parent(s): d415ad5

Update diffusion_lens.py

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