Skip to content

Commit d4b3e74

Browse files
committed
undo indent changes
1 parent 536afaa commit d4b3e74

File tree

1 file changed

+9
-9
lines changed

1 file changed

+9
-9
lines changed

README.md

Lines changed: 9 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -78,9 +78,9 @@ import torch
7878
from chronos import ChronosPipeline
7979

8080
pipeline = ChronosPipeline.from_pretrained(
81-
"amazon/chronos-t5-small",
82-
device_map="cuda", # use "cpu" for CPU inference and "mps" for Apple Silicon
83-
torch_dtype=torch.bfloat16,
81+
"amazon/chronos-t5-small",
82+
device_map="cuda", # use "cpu" for CPU inference and "mps" for Apple Silicon
83+
torch_dtype=torch.bfloat16,
8484
)
8585

8686
df = pd.read_csv("https://raw.githubusercontent.com/AileenNielsen/TimeSeriesAnalysisWithPython/master/data/AirPassengers.csv")
@@ -89,9 +89,9 @@ df = pd.read_csv("https://raw.githubusercontent.com/AileenNielsen/TimeSeriesAnal
8989
# or a left-padded 2D tensor with batch as the first dimension
9090
# forecast shape: [num_series, num_samples, prediction_length]
9191
forecast = pipeline.predict(
92-
context=torch.tensor(df["#Passengers"]),
93-
prediction_length=12,
94-
num_samples=20,
92+
context=torch.tensor(df["#Passengers"]),
93+
prediction_length=12,
94+
num_samples=20,
9595
)
9696
```
9797

@@ -130,9 +130,9 @@ import torch
130130
from chronos import ChronosPipeline
131131

132132
pipeline = ChronosPipeline.from_pretrained(
133-
"amazon/chronos-t5-small",
134-
device_map="cuda",
135-
torch_dtype=torch.bfloat16,
133+
"amazon/chronos-t5-small",
134+
device_map="cuda",
135+
torch_dtype=torch.bfloat16,
136136
)
137137

138138
df = pd.read_csv("https://raw.githubusercontent.com/AileenNielsen/TimeSeriesAnalysisWithPython/master/data/AirPassengers.csv")

0 commit comments

Comments
 (0)