Update README.md
Browse files
README.md
CHANGED
@@ -47,7 +47,7 @@ normed_seqs = (seqs - mean) / std
|
|
47 |
|
48 |
# generate forecast
|
49 |
prediction_length = 96
|
50 |
-
normed_output = model.generate(normed_seqs, max_new_tokens=prediction_length)
|
51 |
output = std * normed_output + mean # rescale the output to the original scale
|
52 |
|
53 |
print(output.shape)
|
|
|
47 |
|
48 |
# generate forecast
|
49 |
prediction_length = 96
|
50 |
+
normed_output = model.generate(normed_seqs, max_new_tokens=prediction_length)
|
51 |
output = std * normed_output + mean # rescale the output to the original scale
|
52 |
|
53 |
print(output.shape)
|