Skip to content

Commit 6a3b8a4

Browse files
committed
Update model_info.py
1 parent 046d3d7 commit 6a3b8a4

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
@@ -134,9 +134,11 @@ def __repr__(self):
134134
category=ModelCategory.BUILTIN,
135135
state=ModelStates.INACTIVE,
136136
model_type="t5",
137-
config_cls="config.Chronos2ForecastingConfig",
138-
model_cls="models.Chronos2Model",
139137
pipeline_cls="pipeline_chronos2.Chronos2Pipeline",
140-
repo_id="amazon/chronos2-2",
138+
repo_id="amazon/chronos-2",
139+
auto_map={
140+
"AutoConfig": "config.Chronos2ForecastingConfig",
141+
"AutoModelForCausalLM": "models.Chronos2Model",
142+
},
141143
),
142144
}

0 commit comments

Comments
 (0)