We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 046d3d7 commit 6a3b8a4Copy full SHA for 6a3b8a4
iotdb-core/ainode/iotdb/ainode/core/model/model_info.py
@@ -134,9 +134,11 @@ def __repr__(self):
134
category=ModelCategory.BUILTIN,
135
state=ModelStates.INACTIVE,
136
model_type="t5",
137
- config_cls="config.Chronos2ForecastingConfig",
138
- model_cls="models.Chronos2Model",
139
pipeline_cls="pipeline_chronos2.Chronos2Pipeline",
140
- repo_id="amazon/chronos2-2",
+ repo_id="amazon/chronos-2",
+ auto_map={
+ "AutoConfig": "config.Chronos2ForecastingConfig",
141
+ "AutoModelForCausalLM": "models.Chronos2Model",
142
+ },
143
),
144
}
0 commit comments