说明文档

kyooni18/chessi-0.1

基于 SAN 走法序列训练的国际象棋下一步走法语言模型。

使用方法

from transformers import AutoTokenizer, AutoModelForCausalLM

repo = "kyooni18/chessi-0.1"
tokenizer = AutoTokenizer.from_pretrained(repo, trust_remote_code=True, use_fast=False)
model = AutoModelForCausalLM.from_pretrained(repo, trust_remote_code=True)

prompt = "e4 e5 Nf3 Nc6 Bb5"
inputs = tokenizer(prompt, return_tensors="pt")
out = model.generate(**inputs, max_new_tokens=1)
new_ids = out[:, inputs["input_ids"].shape[1]:]
print(tokenizer.decode(new_ids[0], skip_special_tokens=True))

kyooni18/chessi-0.1

作者 kyooni18

↓ 15 ♥ 0

创建时间: 2026-02-09 14:03:36+00:00

更新时间: 2026-02-12 23:51:44+00:00

在 Hugging Face 上查看

文件 (14)

.gitattributes

README.md

__init__.py

chessi-webgpu.onnx ONNX

config.json

configuration_chesslm.py

export_metadata.json

generation_config.json

model.safetensors

modeling_chesslm.py

tokenization_chesslm.py

tokenizer_config.json

vocab.json

web_inference_config.json