#codec #encoder #decoder #gpt #bpe

gpt-encoder

Rust BPE 编码器/解码器,用于 GPT-2 / GPT-3

1 个不稳定版本

0.1.1 2023年5月2日
0.1.0 2023年4月24日

#8#bpe

30 每月下载量

MIT 许可证

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