TharunSivamani commited on
Commit
813b8fa
1 Parent(s): 9986ae7
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -85,7 +85,7 @@ def get_output_embeds(input_embeddings):
85
  def generate_with_embs(text_embeddings, seed, max_length):
86
  height = 512 # default height of Stable Diffusion
87
  width = 512 # default width of Stable Diffusion
88
- num_inference_steps = 20 # Number of denoising steps
89
  guidance_scale = 7.5 # Scale for classifier-free guidance
90
  generator = torch.manual_seed(seed) # Seed generator to create the inital latent noise
91
  batch_size = 1
@@ -171,7 +171,7 @@ def generate_image_on_loss(prompt, seed):
171
 
172
  height = 64 # default height of Stable Diffusion
173
  width = 64 # default width of Stable Diffusion
174
- num_inference_steps = 30 # Number of denoising steps
175
  guidance_scale = 8 # Scale for classifier-free guidance
176
  generator = torch.manual_seed(64) # Seed generator to create the inital latent noise
177
  batch_size = 1
@@ -284,4 +284,4 @@ demo = gr.Interface(
284
  ]
285
  )
286
 
287
- demo.launch()
 
85
  def generate_with_embs(text_embeddings, seed, max_length):
86
  height = 512 # default height of Stable Diffusion
87
  width = 512 # default width of Stable Diffusion
88
+ num_inference_steps = 30 # Number of denoising steps
89
  guidance_scale = 7.5 # Scale for classifier-free guidance
90
  generator = torch.manual_seed(seed) # Seed generator to create the inital latent noise
91
  batch_size = 1
 
171
 
172
  height = 64 # default height of Stable Diffusion
173
  width = 64 # default width of Stable Diffusion
174
+ num_inference_steps = 50 # Number of denoising steps
175
  guidance_scale = 8 # Scale for classifier-free guidance
176
  generator = torch.manual_seed(64) # Seed generator to create the inital latent noise
177
  batch_size = 1
 
284
  ]
285
  )
286
 
287
+ demo.launch(debug=True)