LN1996 commited on
Commit
fe0a7d5
1 Parent(s): d6644e5

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -101,7 +101,7 @@ def generate_with_embs(text_embeddings, text_input, seed):
101
 
102
  height = 512 # default height of Stable Diffusion
103
  width = 512 # default width of Stable Diffusion
104
- num_inference_steps = 30 # Number of denoising steps
105
  guidance_scale = 7.5 # Scale for classifier-free guidance
106
  generator = torch.manual_seed(seed) # Seed generator to create the inital latent noise
107
  batch_size = 1
@@ -187,7 +187,7 @@ def generate_with_prompt_style_guidance(prompt, style, seed=42):
187
 
188
  height = 512 # default height of Stable Diffusion
189
  width = 512 # default width of Stable Diffusion
190
- num_inference_steps = 1 # # Number of denoising steps
191
  guidance_scale = 8 # # Scale for classifier-free guidance
192
  generator = torch.manual_seed(seed) # Seed generator to create the inital latent noise
193
  batch_size = 1
 
101
 
102
  height = 512 # default height of Stable Diffusion
103
  width = 512 # default width of Stable Diffusion
104
+ num_inference_steps = 10 # Number of denoising steps
105
  guidance_scale = 7.5 # Scale for classifier-free guidance
106
  generator = torch.manual_seed(seed) # Seed generator to create the inital latent noise
107
  batch_size = 1
 
187
 
188
  height = 512 # default height of Stable Diffusion
189
  width = 512 # default width of Stable Diffusion
190
+ num_inference_steps = 4 # # Number of denoising steps
191
  guidance_scale = 8 # # Scale for classifier-free guidance
192
  generator = torch.manual_seed(seed) # Seed generator to create the inital latent noise
193
  batch_size = 1