Yong99 commited on
Commit
0b4e50a
·
verified ·
1 Parent(s): 788bc4b

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
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)[:, -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)