Update Dockerfile.Public.A10G

Dockerfile.Public.A10G

@@ -11,7 +11,10 @@ RUN pip install https://github.com/camenduru/stable-diffusion-webui-colab/releas
 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 git clone -b v1.6 https://github.com/camenduru/stable-diffusion-webui
+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
+RUN sed -i -e 's/    start()/    #start()/g' /content/stable-diffusion-webui/launch.py
+RUN cd stable-diffusion-webui && python launch.py --skip-torch-cuda-test
 
 ADD https://github.com/camenduru/webui-docker/raw/main/env_patch.py /content/env_patch.py
 RUN sed -i -e '/import image_from_url_text/r /content/env_patch.py' /content/stable-diffusion-webui/modules/ui.py