nkanungo commited on
Commit
4311a33
·
1 Parent(s): 5db5627

Update Era_s20_updt.py

Browse files
Files changed (1) hide show
  1. Era_s20_updt.py +2 -2
Era_s20_updt.py CHANGED
@@ -192,7 +192,7 @@ def get_output_embeds(input_embeddings):
192
  def generate_with_embs_custom(text_embeddings,seed):
193
  height = 512 # default height of Stable Diffusion
194
  width = 512 # default width of Stable Diffusion
195
- num_inference_steps = 10 # Number of denoising steps
196
  guidance_scale = 7.5 # Scale for classifier-free guidance
197
  generator = torch.manual_seed(seed) # Seed generator to create the inital latent noise
198
  batch_size = 1
@@ -278,7 +278,7 @@ class Styles_paintings():
278
  def generate_styles_with_custom_loss(self, image):
279
  height = 512 # default height of Stable Diffusion
280
  width = 512 # default width of Stable Diffusion
281
- num_inference_steps = 20 #@param # Number of denoising steps
282
  guidance_scale = 8 #@param # Scale for classifier-free guidance
283
  batch_size = 1
284
  custom_loss_scale = 200 #@param
 
192
  def generate_with_embs_custom(text_embeddings,seed):
193
  height = 512 # default height of Stable Diffusion
194
  width = 512 # default width of Stable Diffusion
195
+ num_inference_steps = 30 # Number of denoising steps
196
  guidance_scale = 7.5 # Scale for classifier-free guidance
197
  generator = torch.manual_seed(seed) # Seed generator to create the inital latent noise
198
  batch_size = 1
 
278
  def generate_styles_with_custom_loss(self, image):
279
  height = 512 # default height of Stable Diffusion
280
  width = 512 # default width of Stable Diffusion
281
+ num_inference_steps = 10 #@param # Number of denoising steps
282
  guidance_scale = 8 #@param # Scale for classifier-free guidance
283
  batch_size = 1
284
  custom_loss_scale = 200 #@param