Elgato_dark/chronos-forecasting
1
0
Fork 0
mirror of https://github.com/amazon-science/chronos-forecasting synced 2026-05-23 09:39:35 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Remove reference to MPS (#216)

Browse source 
*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 <ansarnd@amazon.de>
This commit is contained in:
Abdul Fatir 2024-11-28 16:05:32 +01:00 committed by GitHub
parent ebaa13cbf0
commit 98b917cc2c
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 1 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
README.md
View file

@ -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,
)