1 个不稳定版本
0.1.1 | 2023年5月2日 |
---|---|
0.1.0 |
|
#8 在 #bpe
30 每月下载量
560KB
176 代码行
GPT-Encoder
Rust BPE 编码器/解码器,用于 GPT-2 / GPT-3
这是对 openai 的 gpt-2 编码器 和 latitudegames 的 GPT-3-Encoder 的 Rust 重新编写
use gpt_encoder::Encoder;
fn main() {
let mut encoder = Encoder::new();
let encoded = encoder.encode("Hello, World".to_string());
println!("{:?}", encoded);
// prints: [15496, 11, 2159]
let decoded = encoder.decode(encoded);
println!("{:?}", decoded);
// prints: "Hello, World"
}
lib.rs
:
GPT-Encoder
Rust BPE 编码器/解码器,用于 GPT-2 / GPT-3
这是对 openai 的 gpt-2 编码器 和 latitudegames 的 GPT-3-Encoder 的 Rust 重新编写
示例
use gpt_encoder::Encoder;
fn main() {
let mut encoder = Encoder::new();
let encoded = encoder.encode("Hello, World".to_string());
println!("{:?}", encoded);
// prints: [15496, 11, 2159]
let decoded = encoder.decode(encoded);
println!("{:?}", decoded);
// prints: "Hello, World"
}
依赖
~2.4–4MB
~69K SLoC