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