Skip to content

Commit 98b917c

Browse files
abdulfatirAbdul Fatir Ansari
andauthored
Remove reference to MPS (#216)
*Issue #, if available:* *Description of changes:* MPS mostly causes issues for users, so let's remove the reference to MPS from the README. Plus, Chronos-Bolt models currently fail on MPS. By submitting this pull request, I confirm that you can use, modify, copy, and redistribute this contribution, under the terms of your choice. Co-authored-by: Abdul Fatir Ansari <[email protected]>
1 parent ebaa13c commit 98b917c

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

README.md

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -104,7 +104,7 @@ from chronos import BaseChronosPipeline
104104

105105
pipeline = BaseChronosPipeline.from_pretrained(
106106
"amazon/chronos-t5-small", # use "amazon/chronos-bolt-small" for the corresponding Chronos-Bolt model
107-
device_map="cuda", # use "cpu" for CPU inference and "mps" for Apple Silicon
107+
device_map="cuda", # use "cpu" for CPU inference
108108
torch_dtype=torch.bfloat16,
109109
)
110110

0 commit comments

Comments
 (0)