开发了一个 VRAM 调度器(vramble),用于在我的本地 AI 工作站上自动切换模型
摘要
作者构建了开源工具 vramble,作为请求调度层自动在本地 AI 工作站上加载/卸载模型并管理队列,解决了多个模型争抢 VRAM 的问题。项目以 MIT 许可证在 GitHub 上开源,欢迎贡献。
我搭建了一台专用工作站,用各种模型和工具进行本地 AI 实验。但它们每一个都想占满所有可用的显存。为了解决这个问题,我开发了一个名为 vramble 的请求调度层。它会自动加载和卸载模型,并通过优先级系统管理队列。现在,我可以运行不同的工作流,无需手动管理模型(也不用浪费 Claude 的 token 来协调这个过程)。欢迎任何贡献,它在我的机器上能用,但毕竟是刚开发出来的,所以无法保证在任何机器上都能直接运行。顺带一提:Qwen 27B 以每秒 200 多个 token 的速度运行,表现比我预期的好得多。opeMIT 许可证,已在 GitHub 上发布:github.com/ValerioDolci/vramble
相似文章
我制作了一个工具,将小型本地模型与大型编码模型串联,并在两者之间自动卸载显存
一位开发者制作了一个工具,将小型本地模型与大型编码模型串联,并在两者之间自动卸载显存以优化内存使用。
我做了一个小应用,可以自动释放显存供游戏使用,并在游戏结束后恢复你的本地 AI 模型。
GamePause 是一个开源的 Windows 托盘小工具,能在检测到游戏启动时自动卸载 LM Studio/Ollama 模型以释放 VRAM,游戏结束后再自动恢复模型。该项目使用 Rust 编写,支持 Steam、Epic、GOG 等平台的自动检测、按游戏自定义规则及崩溃恢复。
我构建了一个工具,用于在8GB GPU上自动交换VRAM,将小型Prompter和大型Coder串联成一个流水线,避免手动切换模型
作者构建了Prompt-Chain,这是一个Streamlit应用,它将一个小型prompter模型和一个大型coder模型串联起来,并自动交换VRAM,从而在8GB GPU上实现高效的代码生成。
本地 AI 多智能体群集
作者探讨了一种面向本地受限硬件(<6GB VRAM)的解耦式多智能体架构:用单个中等规模推理模型作为编排者,配合多个 1-3M 参数的 CPU 端「纳米模型」处理高频例行任务,仅在低置信度时向上升级,实现零云、空气隔离的边缘优先系统。
vLLM 的 RAM 卸载技术 - tcclaviger 致谢帖
感谢 tcclaviger 为 vLLM 添加专家 RAM 卸载支持,使得在本地多 GPU 设置上运行 DeepSeek-V4-Flash-Vision-Exp 等大型 AI 模型成为可能。