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.
2 parents d40eb78 + 752b783 commit 96793ddCopy full SHA for 96793dd
onnx_diagnostic/torch_models/hghub/hub_api.py
@@ -206,6 +206,10 @@ def task_from_id(
206
data = load_architecture_task()
207
if model_id in data:
208
return data[model_id]
209
+ if not config.architectures or not config.architectures:
210
+ # Some hardcoded values until a better solution is found.
211
+ if model_id.startswith("google/bert_"):
212
+ return "fill-mask"
213
assert config.architectures is not None and len(config.architectures) == 1, (
214
f"Cannot return the task of {model_id!r}, pipeline_tag is not setup, "
215
f"architectures={config.architectures} in config={config}. "
0 commit comments