Skip to content

Commit

Permalink
Update optimum/neuron/modeling_diffusion.py
Browse files Browse the repository at this point in the history
Co-authored-by: Michael Benayoun <michael@huggingface.co>
  • Loading branch information
JingyaHuang and michaelbenayoun committed Sep 18, 2024
1 parent 1f15f79 commit fd2a143
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion optimum/neuron/modeling_diffusion.py
Original file line number Diff line number Diff line change
Expand Up @@ -758,7 +758,7 @@ def _export(
configuration files of compatible classes.
unet_id (`Optional[Union[str, Path]]`, defaults to `None`):
A string or a path point to the U-NET model to replace the one in the original pipeline.
token (`Optional[Union[bool,str]]`, defaults to `None`):
token (`Optional[Union[bool, str]]`, defaults to `None`):
The token to use as HTTP bearer authorization for remote files. If `True`, will use the token generated
when running `huggingface-cli login` (stored in `huggingface_hub.constants.HF_TOKEN_PATH`).
revision (`str`, defaults to `"main"`):
Expand Down

0 comments on commit fd2a143

Please sign in to comment.