mirror of
https://github.com/vladmandic/automatic
synced 2026-09-20 01:31:13 +02:00
minor upadtes
This commit is contained in:
Submodule extensions-builtin/sd-dynamic-thresholding updated: c9721ab01a...f02cacfc92
@@ -299,7 +299,7 @@ def parse_prompt_attention(text):
|
||||
square_brackets = []
|
||||
if opts.prompt_attention == 'Fixed attention':
|
||||
res = [[text, 1.0]]
|
||||
log.debug(f'Prompt parse-attention: {opts.prompt_attention} {res}')
|
||||
# log.debug(f'Prompt parse-attention: {opts.prompt_attention} {res}')
|
||||
return res
|
||||
elif opts.prompt_attention == 'Compel parser':
|
||||
conjunction = Compel.parse_prompt_string(text)
|
||||
@@ -308,7 +308,7 @@ def parse_prompt_attention(text):
|
||||
res = []
|
||||
for frag in conjunction.prompts[0].children:
|
||||
res.append([frag.text, frag.weight])
|
||||
log.debug(f'Prompt parse-attention: {opts.prompt_attention} {res}')
|
||||
# log.debug(f'Prompt parse-attention: {opts.prompt_attention} {res}')
|
||||
return res
|
||||
elif opts.prompt_attention == 'A1111 parser':
|
||||
re_attention = re_attention_v1
|
||||
@@ -363,7 +363,7 @@ def parse_prompt_attention(text):
|
||||
res.pop(i + 1)
|
||||
else:
|
||||
i += 1
|
||||
log.debug(f'Prompt parse-attention: {opts.prompt_attention} {res}')
|
||||
# log.debug(f'Prompt parse-attention: {opts.prompt_attention} {res}')
|
||||
return res
|
||||
|
||||
if __name__ == "__main__":
|
||||
|
||||
@@ -349,6 +349,7 @@ def validate_train_inputs(model_name, learn_rate, batch_size, gradient_step, dat
|
||||
|
||||
def train_embedding(id_task, embedding_name, learn_rate, batch_size, gradient_step, data_root, log_directory, training_width, training_height, varsize, steps, clip_grad_mode, clip_grad_value, shuffle_tags, tag_drop_out, latent_sampling_method, use_weight, create_image_every, save_embedding_every, template_filename, save_image_with_stored_embedding, preview_from_txt2img, preview_prompt, preview_negative_prompt, preview_steps, preview_sampler_index, preview_cfg_scale, preview_seed, preview_width, preview_height): # pylint: disable=unused-argument
|
||||
|
||||
shared.log.debug(f'train_embedding: embedding_name={embedding_name}|learn_rate={learn_rate}|batch_size={batch_size}|gradient_step={gradient_step}|data_root={data_root}|log_directory={log_directory}|training_width={training_width}|training_height={training_height}|varsize={varsize}|steps={steps}|clip_grad_mode={clip_grad_mode}|clip_grad_value={clip_grad_value}|shuffle_tags={shuffle_tags}|tag_drop_out={tag_drop_out}|latent_sampling_method={latent_sampling_method}|use_weight={use_weight}|create_image_every={create_image_every}|save_embedding_every={save_embedding_every}|template_filename={template_filename}|save_image_with_stored_embedding={save_image_with_stored_embedding}|preview_from_txt2img={preview_from_txt2img}|preview_prompt={preview_prompt}|preview_negative_prompt={preview_negative_prompt}|preview_steps={preview_steps}|preview_sampler_index={preview_sampler_index}|preview_cfg_scale={preview_cfg_scale}|preview_seed={preview_seed}|preview_width={preview_width}|preview_height={preview_height}')
|
||||
save_embedding_every = save_embedding_every or 0
|
||||
create_image_every = create_image_every or 0
|
||||
template_file = textual_inversion_templates.get(template_filename, None)
|
||||
|
||||
Reference in New Issue
Block a user