Build: Updating to allow passing DOCKER_GPU_ARGS at model generation

[pvijayakrish]

What does the PR do?

Updating the model generation script to allow passing DOCKER_GPU_ARGS directly.

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.
• [x] 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

Related PRs:

Where should the reviewer start?

Test plan:

• CI Pipeline ID:

Caveats:

Background

Related Issues: (use one of the action keywords Closes / Fixes / Resolves / Relates to)

• closes GitHub issue: #xxx