Build: Fixing model generation

[pvijayakrish]

What does the PR do?

Fixing model generation.

Checklist

• [x] PR title reflects the change and is of format <commit_type>: <Title>
• [x] Changes are described in the pull request.
• [ ] Related issues are referenced.
• [ ] Populated github labels field
• [ ] Added test plan and verified test passes.
• [ ] Verified that the PR passes existing CI.
• [ ] Verified copyright is correct on all changed files.
• [ ] Added succinct git squash message before merging ref.
• [ ] All template sections are filled out.
• [ ] Optional: Additional screenshots for behavior/output changes with before/after.

Commit Type:

Check the conventional commit type box here and add the label to the github PR.

• [x] build
• [ ] ci
• [ ] docs
• [ ] feat
• [ ] fix
• [ ] perf
• [ ] refactor
• [ ] revert
• [ ] style
• [ ] test