Skip to content

Commit

Permalink
Update optimum/exporters/neuron/convert.py
Browse files Browse the repository at this point in the history
Co-authored-by: Michael Benayoun <[email protected]>
  • Loading branch information
JingyaHuang and michaelbenayoun authored Mar 27, 2024
1 parent ad44574 commit cb90a0a
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion optimum/exporters/neuron/convert.py
Original file line number Diff line number Diff line change
Expand Up @@ -314,7 +314,7 @@ def export_models(
configs (`Optional[Dict[str, Any]]`, defaults to `None`):
A list of pretrained model configs.
model_name_or_path (`Optional[str]`, defaults to `None`):
Path to pretrained model or model identifier from huggingface.co/models.
Path to pretrained model or model identifier from the Hugging Face Hub.
Returns:
`Tuple[List[List[str]], List[List[str]]]`: A tuple with an ordered list of the model's inputs, and the named
outputs from the Neuron configuration.
Expand Down

0 comments on commit cb90a0a

Please sign in to comment.