Update Dockerfile

Dockerfile

@@ -12,6 +12,7 @@ RUN pip install --pre triton
 RUN pip install numexpr
 
 RUN git clone -b v1.6 https://github.com/camenduru/stable-diffusion-webui && cd stable-diffusion-webui && sed -i -e 's/    start()/    #start()/g' launch.py && python launch.py --skip-torch-cuda-test
+RUN sed -i -e '''/prepare_environment()/a\    os.system\(f\"""sed -i -e ''\"s/dict()))/dict())).cuda()/g\"'' /content/stable-diffusion-webui/repositories/stable-diffusion-stability-ai/ldm/util.py""")''' /content/stable-diffusion-webui/launch.py
 
 # ----------------------------Delete this block if you don't want to see the extra header----------------------------
 ADD https://github.com/camenduru/webui-docker/raw/main/env_patch.py /content/env_patch.py