FROM public.ecr.aws/o7x6j3x6/sd-dreambooth-finetuning-v2


#downgrade diffusers 0.10.0, fix  forward_xformers() got an unexpected keyword argument 'encoder_hidden_states'
RUN sed -i 's/diffusers.pipelines.pipeline_utils/diffusers.pipeline_utils/g' /opt/ml/code/extensions/sd_dreambooth_extension/dreambooth/xattention.py


RUN curl -L https://github.com/peak/s5cmd/releases/download/v2.0.0/s5cmd_2.0.0_Linux-64bit.tar.gz | tar -xz && mv s5cmd /opt/conda/bin/

RUN mkdir -p /opt/ml/input/data/images
COPY images/ /opt/ml/input/data/images/

COPY ./sd_code/ /opt/ml/code/

COPY train.sh /opt/ml/code/
COPY train_dreambooth.py /opt/ml/code/