0 Releases 4 Tags

RSS Feed

• v1.0.0 d2aa6013d7

  Compare

  ## 小狸AI v1.0.0

  shuiyuPyLab released this 2026-06-05 23:10:05 +08:00 | 52 commits to master since this release

  简介

  小狸是一个自研中文对话AI模型，基于Transformer架构。

  模型参数

  • 参数量：52M
  • 架构：d_model=384, n_layers=6, n_heads=8
  • 特性：MoE专家混合 + Thinking思维链 + Memory记忆模块
  • 词表：933个中文字符

  包含文件

  • chat_server.py - Flask网页聊天服务
  • llm_chat_improved.py - 模型架构+推理代码
  • templates/chat.html - 聊天网页UI
  • all_data_cleaned.json - 训练数据（350条）
  • multiturn_data.json - 多轮对话数据
  • xiaoli_data.json - 原始数据

  使用方法

  pip install flask torch
  python chat_server.py
  # 打开 http://127.0.0.1:5000
  

  注意

  • 模型权重文件(xiaoli_final_v4.pth)需单独训练生成
  • 首次运行会自动训练模型

  Downloads

  • Source Code (ZIP)
  • Source Code (TAR.GZ)