From 21e354d9bca86e13ad36953824e9f594ec372286 Mon Sep 17 00:00:00 2001
From: Vladimir Mandic <mandic00@live.com>
Date: Thu, 2 Mar 2023 15:56:46 -0500
Subject: [PATCH] fix lora

---
 cli/modules/util.py                         | 1 +
 cli/train-lora.py                           | 3 ++-
 extensions-builtin/sd-extension-system-info | 2 +-
 requirements.txt                            | 4 +++-
 wiki                                        | 2 +-
 5 files changed, 8 insertions(+), 4 deletions(-)

diff --git a/cli/modules/util.py b/cli/modules/util.py
index 7241b0f83..479b77233 100755
--- a/cli/modules/util.py
+++ b/cli/modules/util.py
@@ -67,6 +67,7 @@ def get_memory():
 
 
 class Map(dict):
+    __slots__ = ('__dict__')
     def __init__(self, *args, **kwargs):
         super(Map, self).__init__(*args, **kwargs)
         for arg in args:
diff --git a/cli/train-lora.py b/cli/train-lora.py
index 2499a1743..88f59ded7 100755
--- a/cli/train-lora.py
+++ b/cli/train-lora.py
@@ -42,7 +42,7 @@ options = Map({
     "train_data_dir": "",
     "shuffle_caption": False,
     "caption_extension": ".txt",
-    "caption_extention": None,
+    "caption_extention": ".txt",
     "keep_tokens": None,
     "color_aug": False,
     "flip_aug": False,
@@ -182,6 +182,7 @@ if __name__ == '__main__':
     Path(dir).mkdir(parents=True, exist_ok=True)
     json_file = os.path.join(tempfile.gettempdir(), args.output, args.output + '.json')
     options.train_data_dir = os.path.join(tempfile.gettempdir(), args.output)
+    res = None
 
     if args.dreambooth:
         options.in_json = None
diff --git a/extensions-builtin/sd-extension-system-info b/extensions-builtin/sd-extension-system-info
index ccb2ac79b..d0783d049 160000
--- a/extensions-builtin/sd-extension-system-info
+++ b/extensions-builtin/sd-extension-system-info
@@ -1 +1 @@
-Subproject commit ccb2ac79b0b6dadc81a38693e87df76febd18544
+Subproject commit d0783d0492f5ea9249d420d207ba4514e54ac4a7
diff --git a/requirements.txt b/requirements.txt
index 11fd06687..1b6b0fb19 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -1,8 +1,8 @@
 accelerate
 aenum
 basicsr
-blendmodes
 bitsandbytes
+blendmodes
 clean-fid
 colormap
 diffusers
@@ -37,7 +37,9 @@ resize-right
 safetensors
 scikit-image
 timm
+toml
 torch
 torchdiffeq
 torchsde
 transformers
+voluptuous
diff --git a/wiki b/wiki
index 5f5b72851..454829722 160000
--- a/wiki
+++ b/wiki
@@ -1 +1 @@
-Subproject commit 5f5b72851dd9442dff7faff74e5bd87c37887ffe
+Subproject commit 45482972278768b7697bd52695fd3f957ff8a9c7