3 个版本 (破坏性)

0.2.0 2022 年 8 月 15 日
0.1.0 2022 年 6 月 16 日
0.0.1 2022 年 6 月 16 日

#783 in 机器学习

每月 26 次下载

Apache-2.0

64KB
1K SLoC

C++ 802 SLoC // 0.2% comments Rust 235 SLoC Python 73 SLoC // 0.3% comments

triton-client-rs

NVIDIA Triton 的 Rust gRPC 客户端库。

此库提供从 NVIDIA 的 Protocol Buffers 定义生成 Triton 客户端所需的设置。

// un-auth'd use of Triton
let client = Client::new("https://127.0.0.1:8001/", None).await?;
let models = client
    .repository_index(triton_client::inference::RepositoryIndexRequest {
        repository_name: "".into(), // This should show us models not referenced by repo name.
        ready: false,               // show all models, not just ready ones.
    })
    .await?;

依赖关系

~14–28MB
~440K SLoC