startup options depending on if cuda is available or not

Dockerfile

@@ -121,4 +121,4 @@ RUN chmod +x on_start.sh
 
 EXPOSE 7860
 
-CMD ["/opt/venv/bin/python", "run.py", "--force-enable-xformers", "--xformers", "--listen", "--enable-insecure-extension-access", "--ui-config-file", "ui-config.json", "--ui-settings-file", "config.json", "--disable-console-progressbars", "--cors-allow-origins", "huggingface.co,hf.space", "--no-progressbar-hiding", "--enable-console-prompts", "--no-download-sd-model", "--api", "--skip-version-check"]
+CMD ["/opt/venv/bin/python", "run.py", "--listen", "--enable-insecure-extension-access", "--ui-config-file", "ui-config.json", "--ui-settings-file", "config.json", "--disable-console-progressbars", "--cors-allow-origins", "huggingface.co,hf.space", "--no-progressbar-hiding", "--enable-console-prompts", "--no-download-sd-model", "--api", "--skip-version-check"]

run.py

@@ -26,4 +26,10 @@ def start():
 
 
 if __name__ == "__main__":
+    import torch
+    if not torch.cuda.is_available():
+        sys.argv.extend(["--precision", "full", "--no-half", "--use-cpu", "SD", "BSRGAN", "ESRGAN", "SCUNet", "CodeFormer", "--all"])
+    else:
+        sys.argv.extend(["--force-enable-xformers", "--xformers"])
+
     start()