Update README.md

abdulfatir · web-flow · commit 7b97a61e0f90 · 2024-04-08T17:25:47.000+02:00
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,
 )
 

Original file line number	Diff line number	Diff line change
`@@ -50,7 +50,7 @@ from chronos import ChronosPipeline`
`50`	`50`
`51`	`51`	`pipeline = ChronosPipeline.from_pretrained(`
`52`	`52`	`"amazon/chronos-t5-small",`
`53`		`- device_map="cuda", # use cpu for CPU inference and mps for Apple Silicon`
	`53`	`+ device_map="cuda", # use "cpu" for CPU inference and "mps" for Apple Silicon`
`54`	`54`	`torch_dtype=torch.bfloat16,`
`55`	`55`	`)`
`56`	`56`