RashiAgarwal commited on
Commit
0cdf49f
1 Parent(s): 1065df4

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -101,7 +101,7 @@ def get_output_embeds(input_embeddings):
101
  def generate_with_embs(text_embeddings, text_input):
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(64) # Seed generator to create the inital latent noise
107
  batch_size = 1
@@ -234,7 +234,7 @@ def inference(prompt, style_index):
234
 
235
  height = 512 # default height of Stable Diffusion
236
  width = 512 # default width of Stable Diffusion
237
- num_inference_steps = 10 # # Number of denoising steps
238
  guidance_scale = 8 # # Scale for classifier-free guidance
239
  generator = torch.manual_seed(64) # Seed generator to create the inital latent noise
240
  batch_size = 1
 
101
  def generate_with_embs(text_embeddings, text_input):
102
  height = 512 # default height of Stable Diffusion
103
  width = 512 # default width of Stable Diffusion
104
+ num_inference_steps = 3 # Number of denoising steps
105
  guidance_scale = 7.5 # Scale for classifier-free guidance
106
  generator = torch.manual_seed(64) # Seed generator to create the inital latent noise
107
  batch_size = 1
 
234
 
235
  height = 512 # default height of Stable Diffusion
236
  width = 512 # default width of Stable Diffusion
237
+ num_inference_steps = 3 # # Number of denoising steps
238
  guidance_scale = 8 # # Scale for classifier-free guidance
239
  generator = torch.manual_seed(64) # Seed generator to create the inital latent noise
240
  batch_size = 1