Update README.md

amazon-science · Apr 8, 2024 · 7b97a61 · 7b97a61
1 parent 45ee467
commit 7b97a61
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/README.md b/README.md
@@ -50,7 +50,7 @@ from chronos import ChronosPipeline
 
 pipeline = ChronosPipeline.from_pretrained(
     "amazon/chronos-t5-small",
-    device_map="cuda",  # use cpu for CPU inference and mps for Apple Silicon
+    device_map="cuda",  # use "cpu" for CPU inference and "mps" for Apple Silicon
     torch_dtype=torch.bfloat16,
 )