From 98b917cc2c372945f2d2b0100cf87a0f1d6180d4 Mon Sep 17 00:00:00 2001 From: Abdul Fatir Date: Thu, 28 Nov 2024 16:05:32 +0100 Subject: [PATCH] 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 --- README.md | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/README.md b/README.md index 7c09b8e..7cf86cc 100644 --- a/README.md +++ b/README.md @@ -104,7 +104,7 @@ from chronos import BaseChronosPipeline pipeline = BaseChronosPipeline.from_pretrained( "amazon/chronos-t5-small", # use "amazon/chronos-bolt-small" for the corresponding Chronos-Bolt model - device_map="cuda", # use "cpu" for CPU inference and "mps" for Apple Silicon + device_map="cuda", # use "cpu" for CPU inference torch_dtype=torch.bfloat16, )