Moves 2 models from fill-mask task to sentence similarity task #455

amd-vivekag · 2025-03-03T09:19:56Z

Moves following 2 models from fill-mask list to new list of sentence-similarity:

hf_multi-qa-mpnet-base-cos-v1
hf_all-mpnet-base-v1

…ew class for these models where tokenizer not required

… hf_wasmai-7b-v1 model from the hf list

…emoves unncessary imports

…padding=False

…is not helping running the model

…for Multiple Choice

Vivek Agrawal and others added 26 commits February 10, 2025 14:55

Skip tokenizer checks in favor of AutoTokenizer

459acff

adds speech processing models to a separate list file, also creates n…

8a43471

…ew class for these models where tokenizer not required

adds try-catch block while exporting the model

98dd771

improves logging of try-catch while exporting model

ce8cf1d

Merge branch 'nod-ai:main' into fix_oom_failure

65545f0

Externalizes large HF models

7ea392d

Merge from main to get latest

f08a8c9

adds try-catch block while exporting the model

32d579e

Externalizes large HF models

38bc7ac

fixes merge issues

685f9d6

Fixes the opset_version issue, OOM issue, input mismatch issue

7b805ae

fixes merge issue for RandomInput class

2541aa2

Fixes opset_version issue, OOM issue for hf_StableBeluga2 and removes…

68d6d38

… hf_wasmai-7b-v1 model from the hf list

removes debug prints

6dd7462

Moves update_no_ext function from framework.py to onnx_utils.py

6739424

creates class function which calls the util function internally

200707f

fixes review comments: export_params during torch.onnx.export call, r…

0a81220

…emoves unncessary imports

Fixes nullptr assert issue by fixing opset_version issue

3284693

Adds docstring to the new function update_model_without_ext_data

7eaf927

removes unnecessary blank line

98c3763

Fixes padding issue by externalizing the models which internally set …

fd35fec

…padding=False

Merge branch 'main' into fix_oom_failure

1935896

fixes large_model call issue

7c62024

uses continum export for hf_StableBeluga2 model as torch.onnx.export …

45eb570

…is not helping running the model

replaces the non-existing models with existing models from Liberty-L …

f9f9990

…for Multiple Choice

Fixes wrong task names for 2 models

03b5604

amd-vivekag requested review from vinayakdsci and zjgarvey and removed request for vinayakdsci March 3, 2025 09:20

amd-vivekag requested a review from pdhirajkumarprasad March 3, 2025 09:20

Provide feedback