From 8151bc57b1dfbde89534ba64a71ac443bb53c332 Mon Sep 17 00:00:00 2001 From: Olivia Godone-Maresca Date: Sat, 15 Jul 2023 12:57:37 -0400 Subject: [PATCH] Format LoRA model/strength as comma separated string in text metadata --- ui/easydiffusion/utils/save_utils.py | 11 +++++++++-- 1 file changed, 9 insertions(+), 2 deletions(-) diff --git a/ui/easydiffusion/utils/save_utils.py b/ui/easydiffusion/utils/save_utils.py index 75d35dc8..d68cf25d 100644 --- a/ui/easydiffusion/utils/save_utils.py +++ b/ui/easydiffusion/utils/save_utils.py @@ -192,7 +192,14 @@ def get_metadata_entries_for_request(req: GenerateImageRequest, task_data: TaskD # if text, format it in the text format expected by the UI is_txt_format = task_data.metadata_output_format and "txt" in task_data.metadata_output_format.lower().split(",") if is_txt_format: - metadata = {TASK_TEXT_MAPPING[key]: val for key, val in metadata.items() if key in TASK_TEXT_MAPPING} + def format_value(value): + if isinstance(value, list): + return ", ".join([ str(it) for it in value ]) + return value + + metadata = { + TASK_TEXT_MAPPING[key]: format_value(val) for key, val in metadata.items() if key in TASK_TEXT_MAPPING + } entries = [metadata.copy() for _ in range(req.num_outputs)] for i, entry in enumerate(entries): @@ -232,7 +239,7 @@ def get_printable_request(req: GenerateImageRequest, task_data: TaskData): used_embeddings.extend(scan_directory(entry.path)) return used_embeddings used_embeddings = scan_directory(os.path.join(app.MODELS_DIR, "embeddings")) - metadata["use_embedding_models"] = ", ".join(used_embeddings) if len(used_embeddings) > 0 else None + metadata["use_embedding_models"] = used_embeddings if len(used_embeddings) > 0 else None # Clean up the metadata if req.init_image is None and "prompt_strength" in metadata: