Skip to content

Commit

Permalink
Add missing comma in example script (#6)
Browse files Browse the repository at this point in the history
*Issue #, if available:*

*Description of changes:*


By submitting this pull request, I confirm that you can use, modify,
copy, and redistribute this contribution, under the terms of your
choice.
  • Loading branch information
abdulfatir authored Mar 12, 2024
1 parent 307b92a commit 1a7bb17
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion README.md
Original file line number Diff line number Diff line change
Expand Up @@ -44,7 +44,7 @@ import torch
from chronos import ChronosPipeline

pipeline = ChronosPipeline.from_pretrained(
"amazon/chronos-t5-small"
"amazon/chronos-t5-small",
device_map="cuda",
torch_dtype=torch.bfloat16,
)
Expand Down

0 comments on commit 1a7bb17

Please sign in to comment.