From fa2a625b92361730e67a059c3840b66735469acd Mon Sep 17 00:00:00 2001 From: Abdul Fatir Ansari Date: Thu, 28 Nov 2024 15:00:47 +0000 Subject: [PATCH] Remove reference to MPS --- README.md | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/README.md b/README.md index 7c09b8ed..7cf86cc0 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, )