diff --git a/README.md b/README.md
index 2a54c24..e978aae 100644
--- a/README.md
+++ b/README.md
@@ -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,
 )