sampler, and similar images in ui

Author: divamgupta

backends/stable_diffusion/diffusionbee_backend.py

@@ -132,6 +132,7 @@ def process_opt(d, generator):
  input_image=d['input_image'],
  tdict_path=tdict_path,
  mode=model_mode,
+ scheduler=d['scheduler'],
  mask_image=d['mask_image'],
  input_image_strength=(float(d['img_strength'])),
  )
@@ -169,6 +170,7 @@ def callback(state="" , progress=-1):
 
  default_d = { "W" : 512 , "H" : 512, "num_imgs":1 , "ddim_steps" : 25 ,
  "scale" : 7.5, "batch_size":1 , "input_image" : None, "img_strength": 0.5
+ , "scheduler":"ddim"
  , "negative_prompt" : "" , "mask_image" : None, "model_id": 0 , "custom_model_path":None}
 
 

backends/stable_diffusion/stable_diffusion.py

@@ -38,7 +38,6 @@
 else:
  from fake_interface import ModelInterface
 
-# from diffusers import LMSDiscreteScheduler, PNDMScheduler, DDIMScheduler
 from schedulers.scheduling_ddim import DDIMScheduler
 from schedulers.scheduling_lms_discrete import LMSDiscreteScheduler
 from schedulers.scheduling_pndm import PNDMScheduler
@@ -357,8 +356,8 @@ def get_unet_out(self, sd_run):
  text_emb_combined = np.concatenate([sd_run.unconditional_context , sd_run.context ])
 
  o = self.model.run_unet(unet_inp=latent_combined, time_emb=temb_combined, text_emb=text_emb_combined )
- sd_run.predicted_unconditional_latent = o[0:1]
- sd_run.predicted_latent = o[1:2]
+ sd_run.predicted_unconditional_latent = o[0: o.shape[0]//2 ]
+ sd_run.predicted_latent = o[o.shape[0]//2 :]
  else:
  sd_run.predicted_unconditional_latent = self.model.run_unet(unet_inp=latent_model_input, time_emb=t_emb, text_emb=sd_run.unconditional_context )
  sd_run.predicted_latent = self.model.run_unet(unet_inp=latent_model_input, time_emb=t_emb, text_emb=sd_run.context)

electron_app/src/components/History.vue

@@ -144,7 +144,7 @@ export default {
 
  get_box_params_dict(box){
  let r = {};
- let vals = {"seed" : "Seed" , "guidence_scale" : "Scale" , "dif_steps":"Steps" , "inp_img_strength" : "Image Strength" , "img_w":"Img Width" , "img_h": "Img Height" , "negative_prompt" : "Negative Prompt" , "model_version":"model_version"}
+ let vals = {"seed" : "Seed" , "guidence_scale" : "Scale" , "dif_steps":"Steps" , "inp_img_strength" : "Image Strength" , "img_w":"Img Width" , "img_h": "Img Height" , "negative_prompt" : "Negative Prompt" , "model_version":"model_version", "selected_sampler":"Sampler" , "use_soft_seed":"Similar Imgs"}
  for(let k in vals)
  if( box[k])
  r[vals[k]] = box[k];

electron_app/src/components/ImgGenerate.vue

@@ -24,7 +24,7 @@
 
  <div class="l_button button_medium button_colored" style="float:right ; " @click="generate_from_prompt">Generate</div>
 
- <SDOptionsDropdown :options_model_values="this_object" :elements_hidden="[ 'inp_img_strength' ]"> </SDOptionsDropdown>
+ <SDOptionsDropdown :options_model_values="this_object" :elements_extra="['sampler' , 'use_soft_seed']"  :elements_hidden="[ 'inp_img_strength' ]"> </SDOptionsDropdown>
 
  <div style="float:right; margin-top: -5px; " >
  <b-dropdown id="dropdown-form" variant="link" ref="dropdown" toggle-class="text-decoration-none" no-caret >
@@ -163,10 +163,12 @@ export default {
  backend_error : "",
  done_percentage : -1,
  is_stopping : false,
+ use_soft_seed:"No",
  modifiers : require("../modifiers.json"),
  is_negative_prompt_avail : false, 
  negative_prompt : "",
- selected_model : 'Default'
+ selected_model : 'Default',
+ selected_sampler : 'ddim',
  };
  
  },
@@ -194,7 +196,8 @@ export default {
  scale : this.guidence_scale , 
  ddim_steps : this.dif_steps, 
  num_imgs : this.num_imgs , 
- batch_size : this.batch_size 
+ batch_size : this.batch_size ,
+ scheduler: this.selected_sampler,
  }
 
  if(this.selected_model && this.selected_model != "Default" && this.app_state.app_data.custom_models[this.selected_model] ){
@@ -205,6 +208,9 @@ export default {
  if(this.is_negative_prompt_avail)
  params['negative_prompt'] = this.negative_prompt;
 
+ if(this.use_soft_seed == 'Yes')
+ params['soft_seed'] = 1
+
  let that = this;
 
  if(this.prompt.trim() == ""){
@@ -226,7 +232,8 @@ export default {
  if(!(that.app_state.app_data.history[history_key])){
  let p = {
  "prompt":that.prompt , "seed": seed, "img_w":that.img_w , "img_h":that.img_h , "key":history_key , "imgs" : [],
- "guidence_scale" : that.guidence_scale , "dif_steps" : that.dif_steps 
+ "guidence_scale" : that.guidence_scale , "dif_steps" : that.dif_steps , "use_soft_seed" : that.use_soft_seed,
+ "selected_sampler" : that.selected_sampler
  }
  if(that.stable_diffusion.model_version)
  p['model_version'] = that.stable_diffusion.model_version;
@@ -279,6 +286,7 @@ export default {
  Seed :this.computed_seed,
  Scale : this.guidence_scale , 
  Steps : this.dif_steps ,
+ sampler_name : "bee" + this.selected_sampler, 
  model_version: this.stable_diffusion.model_version
  }
  if(this.is_negative_prompt_avail)