diff --git a/scripts/txt2img.py b/scripts/txt2img.py index 9d955e3..b38a930 100644 --- a/scripts/txt2img.py +++ b/scripts/txt2img.py @@ -182,7 +182,7 @@ def parse_args(): type=str, help="Device on which Stable Diffusion will be run", choices=["cpu", "cuda"], - default="cpu" + default="cuda" ) parser.add_argument( "--torchscript",