Skip to content

[AINode] Refactor the built-in TimerXL #15655

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
CRZbulabula merged 4 commits into from
Jun 6, 2025
Merged

[AINode] Refactor the built-in TimerXL #15655

Show file tree
Hide file tree
Changes from 1 commit
Commits
Show all changes
4 commits
Select commit Hold shift + click to select a range
ce9b870
almost done
CRZbulabula Jun 5, 2025
fe886e1
seems ok
CRZbulabula Jun 6, 2025
f094cd3
reformat
CRZbulabula Jun 6, 2025
e48fd13
enable revin
CRZbulabula Jun 6, 2025
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Prev Previous commit
Next Next commit
reformat
  • Loading branch information
@CRZbulabula
CRZbulabula committed Jun 6, 2025
commit f094cd32ba6e07393924a654003c2ad74e311e09
1 change: 1 addition & 0 deletions iotdb-core/ainode/ainode/core/manager/inference_manager.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -58,6 +58,7 @@ def infer(self, full_data, predict_length=96, **_):
if data.dtype.byteorder not in ("=", "|"):
data = data.byteswap().newbyteorder()
seqs = torch.tensor(data).unsqueeze(0).float()
# TODO: unify model inference input
output = self.model.generate(seqs, max_new_tokens=predict_length)
Comment thread
CRZbulabula marked this conversation as resolved.
Outdated
df = pd.DataFrame(output[0])
return convert_to_binary(df)
Expand Down
5 changes: 2 additions & 3 deletions iotdb-core/ainode/ainode/core/model/timerxl/modeling_timer.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -33,9 +33,8 @@
)

from ainode.core.log import Logger

from .configuration_timer import TimerConfig
from .ts_generation_mixin import TSGenerationMixin
from ainode.core.model.timerxl.configuration_timer import TimerConfig
from ainode.core.model.timerxl.ts_generation_mixin import TSGenerationMixin

logger = Logger()

Expand Down