From 721aba3f23a09f8e085f365192eff686a8e7ac68 Mon Sep 17 00:00:00 2001 From: Mateus Souza Date: Mon, 5 Jun 2023 12:18:56 -0300 Subject: [PATCH] refact: setting default device as cuda --- scripts/txt2img.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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",