Yaongi/HybriKo-117M-LinuxFC-SFT

ls, cd, mkdir, rm, cp, mv, find, cat, grep,
head, tail, wc, ps, df, du, top, ping, curl,
chmod, tar, Finish

import torch
import sentencepiece as spm
from hybridko.model import HybriKoModel, HybriKoConfig

# Load tokenizer
sp = spm.SentencePieceProcessor()
sp.Load("HybriKo_tok.model")

# Load model
config = HybriKoConfig(
    d_model=768,
    n_layers=12,
    vocab_size=32000,
    n_heads=12,
    n_kv_heads=3,
    ff_mult=3,
    max_seq_len=6144,
    dropout=0.0
)
model = HybriKoModel(config)

checkpoint = torch.load("pytorch_model.pt", map_location="cpu")
state_dict = checkpoint.get("model_state_dict", checkpoint)
model.load_state_dict(state_dict, strict=False)
model.eval()

# Example prompt (ChatML format)
prompt = """<|im_start|>system
You are an AI assistant that helps users execute Linux commands.
<|im_end|>
<|im_start|>user
현재 폴더의 파일 목록을 보여줘
<|im_end|>
<|im_start|>assistant
"""

# Generate
input_ids = torch.tensor([sp.EncodeAsIds(prompt)])
with torch.no_grad():
    output = model.generate(input_ids, max_new_tokens=256)
print(sp.DecodeIds(output[0].tolist()))

@misc{hybridko-linuxfc-2026,
  title={HybriKo-117M-LinuxFC-SFT: Linux Command Function Calling Model},
  author={Yaongi},
  year={2026},
  url={https://huggingface.co/Yaongi/HybriKo-117M-LinuxFC-SFT}
}