mirror of
https://github.com/vladmandic/automatic
synced 2026-09-19 17:24:32 +02:00
fix face scripts
This commit is contained in:
@@ -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]
|
||||
|
||||
@@ -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
|
||||
|
||||
@@ -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])
|
||||
|
||||
Reference in New Issue
Block a user