fix face scripts

This commit is contained in:
Vladimir Mandic
2024-05-31 14:45:55 -04:00
parent 887de74d22
commit fdb5e240c6
4 changed files with 10 additions and 2 deletions
+3 -2
View File
@@ -205,8 +205,9 @@ def face_id(
faceid_model.set_scale(scale)
extra_network_data = None
processing.process_init(p)
p.init(p.all_prompts, p.all_seeds, p.all_subseeds)
if p.all_prompts is None or len(p.all_prompts) == 0:
processing.process_init(p)
p.init(p.all_prompts, p.all_seeds, p.all_subseeds)
for n in range(p.n_iter):
p.iteration = n
p.prompts = p.all_prompts[n * p.batch_size:(n+1) * p.batch_size]
+3
View File
@@ -64,6 +64,9 @@ def instant_id(p: processing.StableDiffusionProcessing, app, source_images, stre
shared.sd_model.to(dtype=devices.dtype)
# pipeline specific args
if p.all_prompts is None or len(p.all_prompts) == 0:
processing.process_init(p)
p.init(p.all_prompts, p.all_seeds, p.all_subseeds)
orig_prompt_attention = shared.opts.prompt_attention
shared.opts.data['prompt_attention'] = 'Fixed attention' # otherwise need to deal with class_tokens_mask
p.task_args['image_embeds'] = face_embeds[0].shape # placeholder
+3
View File
@@ -17,6 +17,9 @@ def photo_maker(p: processing.StableDiffusionProcessing, input_images, trigger,
return None
# validate prompt
if p.all_prompts is None or len(p.all_prompts) == 0:
processing.process_init(p)
p.init(p.all_prompts, p.all_seeds, p.all_subseeds)
trigger_ids = shared.sd_model.tokenizer.encode(trigger) + shared.sd_model.tokenizer_2.encode(trigger)
prompt_ids1 = shared.sd_model.tokenizer.encode(p.all_prompts[0])
prompt_ids2 = shared.sd_model.tokenizer_2.encode(p.all_prompts[0])