JaehaerysNL/StableDiffusion
1
0
Fork 0
mirror of https://github.com/Stability-AI/stablediffusion.git synced 2024-12-22 15:44:58 +00:00
Code Issues Projects Releases Packages Wiki Activity

refact: setting default device as cuda

Browse source
This commit is contained in:
Mateus Souza 2023-06-05 12:18:56 -03:00 committed by GitHub
parent cf1d67a6fd
commit 721aba3f23
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
scripts/txt2img.py
View file

@ -182,7 +182,7 @@ def parse_args():
type=str, type=str,
help="Device on which Stable Diffusion will be run", help="Device on which Stable Diffusion will be run",
choices=["cpu", "cuda"], choices=["cpu", "cuda"],
default="cpu" default="cuda"
) )
parser.add_argument( parser.add_argument(
"--torchscript", "--torchscript",