We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent e4ae9cd commit 795bbbbCopy full SHA for 795bbbb
iotdb-core/ainode/iotdb/ainode/core/model/model_info.py
@@ -136,9 +136,11 @@ def __repr__(self):
136
category=ModelCategory.BUILTIN,
137
state=ModelStates.INACTIVE,
138
model_type="t5",
139
- config_cls="config.Chronos2ForecastingConfig",
140
- model_cls="models.Chronos2Model",
141
pipeline_cls="pipeline_chronos2.Chronos2Pipeline",
142
- repo_id="amazon/chronos2-2",
+ repo_id="amazon/chronos-2",
+ auto_map={
+ "AutoConfig": "config.Chronos2ForecastingConfig",
143
+ "AutoModelForCausalLM": "models.Chronos2Model",
144
+ },
145
),
146
}
0 commit comments