@@ -386,14 +386,14 @@ def get_token_merging_ratio(self, for_hr=False):
386386 return self .token_merging_ratio or opts .token_merging_ratio
387387
388388 def setup_prompts (self ):
389- if type (self .prompt ) == list :
389+ if isinstance (self .prompt , list ) :
390390 self .all_prompts = self .prompt
391- elif type (self .negative_prompt ) == list :
391+ elif isinstance (self .negative_prompt , list ) :
392392 self .all_prompts = [self .prompt ] * len (self .negative_prompt )
393393 else :
394394 self .all_prompts = self .batch_size * self .n_iter * [self .prompt ]
395395
396- if type (self .negative_prompt ) == list :
396+ if isinstance (self .negative_prompt , list ) :
397397 self .all_negative_prompts = self .negative_prompt
398398 else :
399399 self .all_negative_prompts = [self .negative_prompt ] * len (self .all_prompts )
@@ -512,10 +512,10 @@ def __init__(self, p: StableDiffusionProcessing, images_list, seed=-1, info="",
512512 self .s_noise = p .s_noise
513513 self .s_min_uncond = p .s_min_uncond
514514 self .sampler_noise_scheduler_override = p .sampler_noise_scheduler_override
515- self .prompt = self .prompt if type (self .prompt ) != list else self .prompt [0 ]
516- self .negative_prompt = self .negative_prompt if type (self .negative_prompt ) != list else self .negative_prompt [0 ]
517- self .seed = int (self .seed if type (self .seed ) != list else self .seed [0 ]) if self .seed is not None else - 1
518- self .subseed = int (self .subseed if type (self .subseed ) != list else self .subseed [0 ]) if self .subseed is not None else - 1
515+ self .prompt = self .prompt if not isinstance (self .prompt , list ) else self .prompt [0 ]
516+ self .negative_prompt = self .negative_prompt if not isinstance (self .negative_prompt , list ) else self .negative_prompt [0 ]
517+ self .seed = int (self .seed if not isinstance (self .seed , list ) else self .seed [0 ]) if self .seed is not None else - 1
518+ self .subseed = int (self .subseed if not isinstance (self .subseed , list ) else self .subseed [0 ]) if self .subseed is not None else - 1
519519 self .is_using_inpainting_conditioning = p .is_using_inpainting_conditioning
520520
521521 self .all_prompts = all_prompts or p .all_prompts or [self .prompt ]
@@ -741,7 +741,7 @@ def process_images(p: StableDiffusionProcessing) -> Processed:
741741def process_images_inner (p : StableDiffusionProcessing ) -> Processed :
742742 """this is the main loop that both txt2img and img2img use; it calls func_init once inside all the scopes and func_sample once per batch"""
743743
744- if type (p .prompt ) == list :
744+ if isinstance (p .prompt , list ) :
745745 assert (len (p .prompt ) > 0 )
746746 else :
747747 assert p .prompt is not None
@@ -772,12 +772,12 @@ def process_images_inner(p: StableDiffusionProcessing) -> Processed:
772772
773773 p .setup_prompts ()
774774
775- if type (seed ) == list :
775+ if isinstance (seed , list ) :
776776 p .all_seeds = seed
777777 else :
778778 p .all_seeds = [int (seed ) + (x if p .subseed_strength == 0 else 0 ) for x in range (len (p .all_prompts ))]
779779
780- if type (subseed ) == list :
780+ if isinstance (subseed , list ) :
781781 p .all_subseeds = subseed
782782 else :
783783 p .all_subseeds = [int (subseed ) + x for x in range (len (p .all_prompts ))]
@@ -1268,12 +1268,12 @@ def setup_prompts(self):
12681268 if self .hr_negative_prompt == '' :
12691269 self .hr_negative_prompt = self .negative_prompt
12701270
1271- if type (self .hr_prompt ) == list :
1271+ if isinstance (self .hr_prompt , list ) :
12721272 self .all_hr_prompts = self .hr_prompt
12731273 else :
12741274 self .all_hr_prompts = self .batch_size * self .n_iter * [self .hr_prompt ]
12751275
1276- if type (self .hr_negative_prompt ) == list :
1276+ if isinstance (self .hr_negative_prompt , list ) :
12771277 self .all_hr_negative_prompts = self .hr_negative_prompt
12781278 else :
12791279 self .all_hr_negative_prompts = self .batch_size * self .n_iter * [self .hr_negative_prompt ]
0 commit comments