Spaces:
Sleeping
Sleeping
Update diffusion_lens.py
Browse files- 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
|
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')
|