fix lora

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:

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

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