Skip to content

Commit e48fd13

Browse files
committed
enable revin
1 parent f094cd3 commit e48fd13

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

iotdb-core/ainode/ainode/core/manager/inference_manager.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -59,7 +59,7 @@ def infer(self, full_data, predict_length=96, **_):
5959
data = data.byteswap().newbyteorder()
6060
seqs = torch.tensor(data).unsqueeze(0).float()
6161
# TODO: unify model inference input
62-
output = self.model.generate(seqs, max_new_tokens=predict_length)
62+
output = self.model.generate(seqs, max_new_tokens=predict_length, revin=True)
6363
df = pd.DataFrame(output[0])
6464
return convert_to_binary(df)
6565

0 commit comments

Comments
 (0)