Skip to content

Commit c2916aa

Browse files
committed
fix
1 parent 805107a commit c2916aa

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

onnx_diagnostic/torch_models/hghub/model_inputs.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -95,6 +95,8 @@ def get_untrained_model_with_inputs(
9595
print("-- dynamic shapes:", pprint.pformat(data['dynamic_shapes']))
9696
print("-- configuration:", pprint.pformat(data['configuration']))
9797
"""
98+
if task == "":
99+
task = None
98100
assert not use_preinstalled or not use_only_preinstalled, (
99101
f"model_id={model_id!r}, preinstalled model is only available "
100102
f"if use_only_preinstalled is False."

0 commit comments

Comments
 (0)