search

siutin / stable-diffusion-webui-docker

stable-diffusion-webui in docker
https://hub.docker.com/r/siutin/stable-diffusion-webui-docker
105 stars 20 forks source link

training not running after installing the Dreambooth extension #23

Open mozily opened 1 month ago

mozily commented 1 month ago

this method worked well until 2024-10-22, but it suddenly stopped working after that.

image : siutin/stable-diffusion-webui-docker:latest-cuda command : --listen --enable-insecure-extension-access

Until then, I was using it well, but suddenly when I run the train in Dreambooth, it outputs the following error.

Initializing dreambooth training... Traceback (most recent call last): File "/app/stable-diffusion-webui/extensions/sd_dreambooth_extension/dreambooth/ui_functions.py", line 732, in start_training from dreambooth.train_dreambooth import main # noqa ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ File "/app/stable-diffusion-webui/extensions/sd_dreambooth_extension/dreambooth/train_dreambooth.py", line 34, in from diffusers.training_utils import unet_lora_state_dict File "/app/stable-diffusion-webui/venv/lib/python3.11/site-packages/diffusers/training_utils.py", line 27, in if transformers.integrations.deepspeed.is_deepspeed_zero3_enabled(): ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ AttributeError: module 'transformers.integrations' has no attribute 'deepspeed'

During handling of the above exception, another exception occurred:

Traceback (most recent call last): File "/app/stable-diffusion-webui/extensions/sd_dreambooth_extension/dreambooth/ui_functions.py", line 734, in start_training from dreambooth.train_dreambooth import main # noqa ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ File "/app/stable-diffusion-webui/extensions/sd_dreambooth_extension/dreambooth/train_dreambooth.py", line 34, in from diffusers.training_utils import unet_lora_state_dict File "/app/stable-diffusion-webui/venv/lib/python3.11/site-packages/diffusers/training_utils.py", line 27, in if transformers.integrations.deepspeed.is_deepspeed_zero3_enabled(): ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ AttributeError: module 'transformers.integrations' has no attribute 'deepspeed' Duration: 00:00:00 Duration: 00:00:01

Please help me understand why this error is occurring.

siutin commented 4 weeks ago

Sorry to hear about that, currently there have been changes in the way releases are related to the CUDA version. Previously, each release was tied to CUDA version 12.1.1. Could you try using this to see if it works?

image: siutin/stable-diffusion-webui-docker:latest-cuda-12.1.1

mozily commented 4 weeks ago

@siutin thank you for the quick response.

I tried again using the "latest-cuda-12.1.1" image you mentioned, but the same error as in the main text is displayed.

The last image I successfully used was the "latest-cuda" image from 2024-10-22. With the same image, I was able to recreate the container, install DreamBooth, and even build the model. However, when I attempt to train it afterward, the same error as above is displayed.