From 0e10484fdd3f0dbb9be405a478fcb40c342835c8 Mon Sep 17 00:00:00 2001 From: Bruno Rousseau Date: Sat, 19 Oct 2024 07:22:54 -0400 Subject: [PATCH] Fix bug. --- crystal_diffusion/sample_diffusion.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/crystal_diffusion/sample_diffusion.py b/crystal_diffusion/sample_diffusion.py index 81f09e8c..ea598c63 100644 --- a/crystal_diffusion/sample_diffusion.py +++ b/crystal_diffusion/sample_diffusion.py @@ -47,7 +47,7 @@ def main(args: Optional[Any] = None): "--output", required=True, help="path to outputs - will store files here" ) parser.add_argument( - "--device", default="gpu", help="Device to use. Defaults to cuda." + "--device", default="cuda", help="Device to use. Defaults to cuda." ) args = parser.parse_args(args) if os.path.exists(args.output):