Skip to content

Commit 795bbbb

Browse files
committed
Update model_info.py
1 parent e4ae9cd commit 795bbbb

File tree

1 file changed

+5
-3
lines changed

1 file changed

+5
-3
lines changed

iotdb-core/ainode/iotdb/ainode/core/model/model_info.py

Lines changed: 5 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -136,9 +136,11 @@ def __repr__(self):
136136
category=ModelCategory.BUILTIN,
137137
state=ModelStates.INACTIVE,
138138
model_type="t5",
139-
config_cls="config.Chronos2ForecastingConfig",
140-
model_cls="models.Chronos2Model",
141139
pipeline_cls="pipeline_chronos2.Chronos2Pipeline",
142-
repo_id="amazon/chronos2-2",
140+
repo_id="amazon/chronos-2",
141+
auto_map={
142+
"AutoConfig": "config.Chronos2ForecastingConfig",
143+
"AutoModelForCausalLM": "models.Chronos2Model",
144+
},
143145
),
144146
}

0 commit comments

Comments
 (0)