hako-mikan / sd-webui-traintrain

LoRA training extention for Stable Diffusion Web-UI
GNU Affero General Public License v3.0
131 stars 4 forks source link

error "mask" #6

Closed ai-is-in-wonderland closed 5 months ago

ai-is-in-wonderland commented 5 months ago

raceback (most recent call last): File "C:\stablediffusion20230811\stable-diffusion-webui\extensions\sd-webui-traintrain\trainer\train.py", line 181, in train_main result = train_lora(t) File "C:\stablediffusion20230811\stable-diffusion-webui\extensions\sd-webui-traintrain\trainer\train.py", line 234, in train_lora for batch in t.dataloader: File "C:\stablediffusion20230811\stable-diffusion-webui\extensions\sd-webui-traintrain\trainer\dataset.py", line 52, in next return next(self.iterators[idx]) File "C:\stablediffusion20230811\stable-diffusion-webui\venv\lib\site-packages\torch\utils\data\dataloader.py", line 633, in next data = self._next_data() File "C:\stablediffusion20230811\stable-diffusion-webui\venv\lib\site-packages\torch\utils\data\dataloader.py", line 677, in _next_data data = self._dataset_fetcher.fetch(index) # may raise StopIteration File "C:\stablediffusion20230811\stable-diffusion-webui\venv\lib\site-packages\torch\utils\data_utils\fetch.py", line 54, in fetch return self.collate_fn(data) File "C:\stablediffusion20230811\stable-diffusion-webui\venv\lib\site-packages\torch\utils\data_utils\collate.py", line 265, in default_collate return collate(batch, collate_fn_map=default_collate_fn_map) File "C:\stablediffusion20230811\stable-diffusion-webui\venv\lib\site-packages\torch\utils\data_utils\collate.py", line 127, in collate return elem_type({key: collate([d[key] for d in batch], collate_fn_map=collate_fn_map) for key in elem}) File "C:\stablediffusion20230811\stable-diffusion-webui\venv\lib\site-packages\torch\utils\data_utils\collate.py", line 127, in return elem_type({key: collate([d[key] for d in batch], collate_fn_map=collate_fn_map) for key in elem}) File "C:\stablediffusion20230811\stable-diffusion-webui\venv\lib\site-packages\torch\utils\data_utils\collate.py", line 127, in return elem_type({key: collate([d[key] for d in batch], collate_fn_map=collate_fn_map) for key in elem}) KeyError: 'mask'

Loss EMA * 1000: 50.6757, Current LR: 1.00e-04, 1.00e-04, Epoch: 0: 1%|▏ | 2/200 [00:02<04:12, 1.27s/it] Loading weights [54ef3e3610] from C:\stablediffusion20230621\stable-diffusion-webui\models\Stable-diffusion\anime\meinamix_meinaV11.safetensors Creating model from config: C:\stablediffusion20230811\stable-diffusion-webui\configs\v1-inpainting-inference.yaml Loading VAE weights specified in settings: C:\stablediffusion20230621\stable-diffusion-webui\models\VAE\vae-ft-mse-840000-ema-pruned.safetensors Applying attention optimization: xformers... done. Model loaded in 1.6s (unload existing model: 0.2s, load weights from disk: 0.1s, create model: 0.2s, apply weights to model: 0.8s, load VAE: 0.2s).

hako-mikan commented 5 months ago

Are you mixing transparent and non-transparent images in the dataset?

hako-mikan commented 5 months ago

OK. I could reproduce it. Thanks.

hako-mikan commented 5 months ago

Fixed.

ai-is-in-wonderland commented 5 months ago

thank you!