
TinyEngram
给大模型装上长期记忆,让对话和生成更连贯
项目预览
项目数据
项目介绍
Star 增长趋势
本站每日记录该项目 GitHub 星标变化,观测越久曲线越完整
数据来源:GitHub API 逐日快照 · 收录于 2026-08-09 · 当前星标 ★ 1343
技术标签
使用教程
ℹ 本教程由 AI 依据该项目 README 整理,个别步骤可能需要按你的环境微调。
环境要求
- 已安装 conda(Miniconda/Anaconda),可用 conda create 创建环境
- Python 3.10(README 指定 python=3.10)
- pip 可用并能联网安装 requirements.txt 中的依赖
- 训练/视觉复现需要 CUDA 环境,具体版本见 doc/reproduction/environment.md
安装与启动步骤
-
1创建 conda 环境
用 Python 3.10 新建名为 tinyengram 的独立环境,-y 表示自动确认,避免污染系统环境。
conda create -n tinyengram python=3.10 -y -
2激活环境
切换到刚创建的环境,后续所有安装与命令都要在该环境下执行。
conda activate tinyengram -
3升级 pip
先把 pip 升到较新版本,减少安装 requirements.txt 时的依赖解析问题。
pip install --upgrade pip -
4安装项目依赖
在仓库根目录执行,安装 README 提供的固定依赖文件;网络慢可加国内镜像源。
pip install -r requirements.txt -
5查看环境与复现文档
README 提示 CUDA 注意事项与可选评估依赖都写在这里,按需补齐后再跑实验。
cat doc/reproduction/environment.md
如何确认成功
无报错装完 requirements.txt,conda activate tinyengram 后可正常导入依赖,即环境就绪;README 未给出更多启动验证方式。
常见问题
Q:CUDA 或评估相关的额外依赖怎么装?
A:README 明确指向 doc/reproduction/environment.md,其中包含 CUDA 注意事项和可选评估依赖说明,按该文档补齐即可。
Q:Python 版本可以用 3.11 或更高吗?
A:README 只给出 python=3.10 的创建命令,为减少依赖冲突建议严格使用 3.10,未验证的版本可能装不上。
Q:安装很慢或超时怎么办?
A:pip install 可加国内镜像源加速,例如 -i https://pypi.tuna.tsinghua.edu.cn/simple;conda 创建慢可换用 Miniconda。
Q:这个项目能直接推理出图吗?
A:TinyEngram 是研究项目,README 主旨是复现 Engram 记忆注入并扩展到 Stable Diffusion 概念注入,需按仓库实验脚本和实验报告自行运行。
注意事项
- 这是研究/复现型项目,README 只提供环境安装步骤,未给出统一的启动或推理命令,训练与实验请结合 doc/reproduction/environment.md 与 doc/experiments 下的实验报告。
- 依赖已固定在 requirements.txt 中,不要随意升级 torch 等核心包版本,以免破坏 pinned 依赖兼容性。
- LoRA/PEFT 微调与 Stable Diffusion 视觉实验对显存有要求,建议在带 CUDA 的 GPU 机器上操作。
- 安装与实验过程较长,建议在 tmux 或 nohup 中执行,避免 SSH 断开导致中断。
核心亮点
- 基于 DeepSeek Engram 架构,前沿记忆注入方案,有明确研究价值
- 支持 Qwen-3 和 Stable Diffusion 多模态结合,扩展性强
- 使用 LoRA/PEFT 高效微调,资源占用相对友好
不足之处
- 早期项目,文档和社区支持待观察
- 实验性质强,生产环境稳定性未验证
适用场景
- 学术研究:探索 LLM 记忆增强机制
- 长对话应用:提升聊天机器人的上下文保持能力
- 多模态生成:结合图像生成与文本记忆
替代项目
MemGPT、LangChain Memory、Hugging Face PEFT
上一篇:dpm-solver
同类项目推荐
AMD-LLM
开源
在 AMD GPU 上快速跑通 1.35 亿参数语言模型训练
Training code and resources for AMD-135M language models on AMD GPUs.
higgsfield
开源
让万亿参数模型训练不再怕GPU故障,自动容错编排
Fault-tolerant, highly scalable GPU orchestration, and a machine learning framework ···
Kardia-R1
开源
让AI学会共情,用强化学习提升情感支持回复质量
[WWW] Kardia-R1: Unleashing LLMs to Reason toward Understanding and Empathy for Emot···
burn-onnx
开源
把 ONNX 模型转成纯 Rust Burn 代码,推理微调不再依赖 Python
Convert ONNX models into native, backend-agnostic Burn code for inference and fine-t···
RAGEN
开源
多轮智能体强化学习训练与推理退化诊断
Agent RL framework for LLM agents: multi-turn reinforcement learning with StarPO and···
LightRFT
开源
用奖励模型驱动微调,让 LLM/VLM 按目标越训越强
LightRFT: Light, Efficient, Omni-modal & Reward-model Driven Reinforcement Fine-Tuni···
gemma-tuner-multimodal
开源
Mac 上直接微调 Gemma 多模态模型,图文音一起训
Fine-tune Gemma 4 and 3n with audio, images and text on Apple Silicon, using PyTorch···
makeMoE
开源
手写稀疏混合专家模型,彻底搞懂 MoE 原理
From scratch implementation of a sparse mixture of experts language model inspired b···
diffusion_distiller
开源
把扩散模型蒸馏成少步采样,生成速度提升数倍
PyTorch Implementation of "Progressive Distillation for Fast Sampling of Diffusion M···