Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
fixes --library-name paramater in example for sentence_transformers.mdx
Looks like we stopped passing in --library-name at some point. This is the only place I could find it in the docs. We infer it from the model type. If I run the example: ``` optimum-cli export neuron -m Snowflake/snowflake-arctic-embed-l --library-name sentence_transformers --sequence_length 1024 --batch_size 1 --task feature-extraction snowflake-arctic-embed-l-inf2 /opt/aws_neuron_venv_pytorch/lib/python3.8/site-packages/diffusers/models/transformers/transformer_2d.py:34: FutureWarning: `Transformer2DModelOutput` is deprecated and will be removed in version 1.0.0. Importing `Transformer2DModelOutput` from `diffusers.models.transformer_2d` is deprecated and this will be removed in a future version. Please use `from diffusers.models.modeling_outputs import Transformer2DModelOutput`, instead. deprecate("Transformer2DModelOutput", "1.0.0", deprecation_message) usage: optimum-cli Optimum CLI tool: error: unrecognized arguments: --library-name snowflake-arctic-embed-l-inf2 ``` If I remove --library-name, the example compiles. In the code example, they show it being passed in. I assume that still works because I see it being used in the tests.
- Loading branch information