Skip to content

Updates to main.py to expose more configuration options to user#12

Merged
EmilyWebber merged 1 commit intomainfrom
anindam
Mar 22, 2026
Merged

Updates to main.py to expose more configuration options to user#12
EmilyWebber merged 1 commit intomainfrom
anindam

Conversation

@amanocha
Copy link
Copy Markdown
Contributor

Issue #, if available:

Description of changes:

By submitting this pull request, I confirm that you can use, modify, copy, and redistribute this contribution, under the terms of your choice.

@neuron-internal-workflow
Copy link
Copy Markdown

🔍 Internal Workflow Results

The check passed with scan ID: a5fbff07-dbc9-4ed8-b6de-a2e67e4c509b

📊 View Detailed Results

Note: The workflow is currently running in shadow mode for testing and validation. Results are reported for visibility, but will not block PR merges.

Comment thread main.py
parser.add_argument("--attention-dtype", type=to_torch_dtype)
parser.add_argument("--output-logits", action="store_true")
parser.add_argument("--vocab-parallel", action="store_true")
parser.add_argument("--skip-compile", type=bool, default=False)
Copy link
Copy Markdown
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

We will need to keep these parameters.

Copy link
Copy Markdown
Contributor Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

See lines 99, 179, and 255

@EmilyWebber EmilyWebber merged commit 43adb97 into main Mar 22, 2026
2 checks passed
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants