escape brackets in lora random prompt generator

2024-12-29 02:45:05 +08:00 · 2024-03-21 16:29:51 +04:00 · 2024-03-21 16:29:51 +04:00 · 721c4309c2
commit 721c4309c2
parent 57727e554d
1 changed files with 2 additions and 0 deletions
--- a/extensions-builtin/Lora/ui_edit_user_metadata.py
+++ b/extensions-builtin/Lora/ui_edit_user_metadata.py
@ -149,6 +149,8 @@ class LoraUserMetadataEditor(ui_extra_networks_user_metadata.UserMetadataEditor)

            v = random.random() * max_count
            if count > v:
+                for x in "({[]})":
+                    tag = tag.replace(x, '\\' + x)
                res.append(tag)

        return ", ".join(sorted(res))