problem with creating new models in Dreambooth extention

[pedaraf]

recently I haven't been able to create model using Dreambooth extention. when I try I get something like the following statements in the console:

Extracting config from /content/lite_colab/extensions/sd_dreambooth_extension/dreambooth/../configs/v2-training-unfrozen.yaml
Extracting checkpoint from /content/lite_colab/models/Stable-diffusion/model.safetensors
[2023-11-01 10:36:37,381][DEBUG][dreambooth.dataclasses.db_config] - Saving to /content/lite_colab/models/dreambooth/appMo
[2023-11-01 10:36:41,265][DEBUG][filelock] - Attempting to acquire lock 135432342704560 on /root/.cache/huggingface/hub/models--CompVis--stable-diffusion-safety-checker/blobs/aa454d222558d7e095fc9343266efd9512ad2f19.lock
[2023-11-01 10:36:41,265][DEBUG][filelock] - Lock 135432342704560 acquired on /root/.cache/huggingface/hub/models--CompVis--stable-diffusion-safety-checker/blobs/aa454d222558d7e095fc9343266efd9512ad2f19.lock
Downloading (…)lve/main/config.json: 100% 4.55k/4.55k [00:00<00:00, 17.5MB/s]
[2023-11-01 10:36:41,488][DEBUG][filelock] - Attempting to release lock 135432342704560 on /root/.cache/huggingface/hub/models--CompVis--stable-diffusion-safety-checker/blobs/aa454d222558d7e095fc9343266efd9512ad2f19.lock
[2023-11-01 10:36:41,488][DEBUG][filelock] - Lock 135432342704560 released on /root/.cache/huggingface/hub/models--CompVis--stable-diffusion-safety-checker/blobs/aa454d222558d7e095fc9343266efd9512ad2f19.lock
[2023-11-01 10:36:43,082][DEBUG][filelock] - Attempting to acquire lock 135432340559408 on /root/.cache/huggingface/hub/models--CompVis--stable-diffusion-safety-checker/blobs/64b8393f1afd5a0c1ed2aa5f341fa7c08286839a48f3743162a76a2835c808bd.lock
[2023-11-01 10:36:43,082][DEBUG][filelock] - Lock 135432340559408 acquired on /root/.cache/huggingface/hub/models--CompVis--stable-diffusion-safety-checker/blobs/64b8393f1afd5a0c1ed2aa5f341fa7c08286839a48f3743162a76a2835c808bd.lock
Downloading pytorch_model.bin: 100% 1.22G/1.22G [00:16<00:00, 72.6MB/s]
[2023-11-01 10:36:59,967][DEBUG][filelock] - Attempting to release lock 135432340559408 on /root/.cache/huggingface/hub/models--CompVis--stable-diffusion-safety-checker/blobs/64b8393f1afd5a0c1ed2aa5f341fa7c08286839a48f3743162a76a2835c808bd.lock
[2023-11-01 10:36:59,967][DEBUG][filelock] - Lock 135432340559408 released on /root/.cache/huggingface/hub/models--CompVis--stable-diffusion-safety-checker/blobs/64b8393f1afd5a0c1ed2aa5f341fa7c08286839a48f3743162a76a2835c808bd.lock
[2023-11-01 10:37:07,598][DEBUG][filelock] - Attempting to acquire lock 135432344436256 on /root/.cache/huggingface/hub/models--CompVis--stable-diffusion-safety-checker/blobs/5294955ff7801083f720b34b55d0f1f51313c5c5.lock
[2023-11-01 10:37:07,598][DEBUG][filelock] - Lock 135432344436256 acquired on /root/.cache/huggingface/hub/models--CompVis--stable-diffusion-safety-checker/blobs/5294955ff7801083f720b34b55d0f1f51313c5c5.lock
Downloading (…)rocessor_config.json: 100% 342/342 [00:00<00:00, 625kB/s]
[2023-11-01 10:37:07,825][DEBUG][filelock] - Attempting to release lock 135432344436256 on /root/.cache/huggingface/hub/models--CompVis--stable-diffusion-safety-checker/blobs/5294955ff7801083f720b34b55d0f1f51313c5c5.lock
[2023-11-01 10:37:07,825][DEBUG][filelock] - Lock 135432344436256 released on /root/.cache/huggingface/hub/models--CompVis--stable-diffusion-safety-checker/blobs/5294955ff7801083f720b34b55d0f1f51313c5c5.lock
/usr/local/lib/python3.10/dist-packages/transformers/models/clip/feature_extraction_clip.py:28: FutureWarning: The class CLIPFeatureExtractor is deprecated and will be removed in version 5 of Transformers. Please use CLIPImageProcessor instead.
warnings.warn(
[2023-11-01 10:39:07,719][DEBUG][dreambooth.utils.model_utils] - Restored system models.
[2023-11-01 10:39:08,445][DEBUG][dreambooth.utils.model_utils] - Restored system models.
Duration: 00:02:34

and when I try to train the created model I see the following exception which I think is related to the created model's issues :

Loading unet...: 80% 4/5 [00:06<00:01, 1.79s/it]Error verifying pickled file from /content/lite_colab/models/dreambooth/qww/working/unet/diffusion_pytorch_model.bin:
Traceback (most recent call last):
File "/content/lite_colab/modules/safe.py", line 136, in load_with_extra
check_pt(filename, extra_handler)
File "/content/lite_colab/modules/safe.py", line 83, in check_pt
check_zip_filenames(filename, z.namelist())
File "/content/lite_colab/modules/safe.py", line 75, in check_zip_filenames
raise Exception(f"bad file inside {filename}: {name}")
Exception: bad file inside /content/lite_colab/models/dreambooth/qww/working/unet/diffusion_pytorch_model.bin: diffusion_pytorch_model/byteorder

The file may be malicious, so the program is not going to read it.
You can skip this check with --disable-safe-unpickle commandline argument.

Traceback (most recent call last):
File "/content/lite_colab/extensions/sd_dreambooth_extension/dreambooth/ui_functions.py", line 730, in start_training
result = main(class_gen_method=class_gen_method)
File "/content/lite_colab/extensions/sd_dreambooth_extension/dreambooth/train_dreambooth.py", line 1809, in main
return inner_loop()
File "/content/lite_colab/extensions/sd_dreambooth_extension/dreambooth/memory.py", line 126, in decorator
return function(batch_size, grad_size, prof, *args, **kwargs)
File "/content/lite_colab/extensions/sd_dreambooth_extension/dreambooth/train_dreambooth.py", line 380, in inner_loop
unet = UNet2DConditionModel.from_pretrained(
File "/usr/local/lib/python3.10/dist-packages/diffusers/models/modeling_utils.py", line 644, in from_pretrained
missing_keys = set(model.state_dict().keys()) - set(state_dict.keys())
AttributeError: 'NoneType' object has no attribute 'keys'
Loading unet...: 80% 4/5 [00:06<00:01, 1.57s/it]
Duration: 00:00:10
[2023-10-30 19:29:30,947][DEBUG][dreambooth.utils.model_utils] - Restored system models.
Duration: 00:00:11