4.2.1.5.2 OML4Pyスリム・クライアントのインストールの確認

OML4Pyスリム・クライアントのインストールを確認します。

OML4Pyスリム・クライアントが正しくインストールされていることを確認するために、ONNXモデル変換ユーティリティのロードなどの主要な機能をテストできます。

次の例では、oml.utilsからONNXモデル変換メソッドをロードし、パイプラインを構成し、モデルをONNXファイルとしてエクスポートする方法を示しています:

from oml.utils import ONNXPipeline, ONNXPipelineConfig, MiningFunction

# Initialize the pipeline with a pre-trained model
em = ONNXPipeline(model_name="sentence-transformers/all-MiniLM-L6-v2")
em.export2file("your_preconfig_file_name", output_dir="my_dir")

# Configure the pipeline using a template and export a different model
config = ONNXPipelineConfig.from_template("text", max_seq_length=512)
em = ONNXPipeline(model_name="intfloat/e5-small-v2", config=config)
em.export2file("your_template_file_name", output_dir="my_dir")

コードを正常に実行できる場合は、OML4Pyスリム・クライアント・インストールがONNXモデル変換に対応していることが確認されました。