
【免费下载链接】DreamDojoOfficial Codebase for DreamDojo: A Generalist Robot World Model from Large-Scale Human Videos (ICML 2026)项目地址https://gitcode.com/gh_mirrors/dr/DreamDojo点击查看免费下载DreamDojo 是一个从 4.4 万小时人类第一视角视频中训练出来的通用机器人世界模型Generalist Robot World ModelICML 2026 官方代码库。它最吸引眼球的能力是经过蒸馏后模型可以一边接收机器人动作动作条件一边实时逐帧生成未来视频——稳定在 10 FPS 持续输出 1 分钟以上并且支持 LeapMotion 手势遥操作输入。本文带你快速上手这套边动边看的实时交互推理并讲清楚它背后的 KV Cache 流式生成原理。一、实时交互世界模型是什么普通视频生成模型是离线渲染给它提示词等几分钟后输出一段完整视频。而 DreamDojo 的交互模式完全不同动作作为输入每一小段时间chunk系统从动作源取一批动作机械臂末端位姿/关节指令维度 384逐块生成模型只生成这一小段对应的 latent 帧解码成像素后立即显示滚动更新上一段的最后一帧自动成为下一段的条件帧视频就这样无限延伸下去。核心逻辑位于实时推理脚本 action_video2world_teleop.py 的run_realtime_generation方法中它就是一个取动作 → 生成一个 chunk → 解码显示 → 更新条件帧的循环。二、为什么能快三阶段蒸馏是前提 ⚡10 FPS 不是靠单卡硬堆算力而是蒸馏流水线把慢学生练出来的。整个流程见官方文档 docs/DISTILL.md阶段作用启动脚本Teacher Generation用教师模型预生成多步去噪目标launch_teacher_gen.shWarmup训练因果学生网络对齐教师输出launch_warmup.shSelf-Forcing用学生自己的自回归预测微调消除长序列误差累积launch_self_forcing.sh蒸馏完成后学生模型每帧只需4 步去噪--num_steps 4这就是实时性的第一块基石。三、KV Cache 流式生成原理 流式生成的关键在 kv_cache.py。它给 DiT 的注意力层包了一个K/V 缓存外壳可以理解为把大语言模型里 Transformer 推理的 KV Cache 思路搬到了视频扩散模型上按 chunk 缓存每个 latent 帧对应一组 H×W 的 token已生成帧的 K/V 直接复用新 chunk 只需计算增量注意力不用重算整段视频滚动窗口max_cache_size以latent 帧数为容量单位超出后丢弃最老的帧保证显存不随时长无限增长无状态stateless模式开启--compile_model时KV 状态不存放在模块内部而是作为张量参数传入传出TensorKVCacheLayerState让torch.compile可以完整追踪计算图、零断图进一步提速。实时类RealtimeStreamingInference继承自 action_video2world.py 中的ActionStreamingInference复用了后者的 TE 算子替换、按块编译、VAE 编译等全套加速配置。四、动作从哪来三种输入源 ️推理脚本内置三种动作源通过--action_source切换见 参数定义random随机游走动作适合快速验证管线file从.npy文件顺序读取真实遥操作动作轨迹用完自动重复最后一帧leapmotionLeapMotion 手势输入。需要注意仓库中该源目前是占位实现——尚未接入 LeapMotion SDK 时会自动回退到随机动作代码里留有明确的 TODO 接入点LeapMotionActionSource。也就是说遥操作接口已经就位你只需把 SDK 数据填进get_actions即可。五、一键运行实时推理最快启动步骤 1. 环境准备按照 docs/SETUP.md 安装环境仓库提供 install.sh 一键脚本并从 datasets/README.md 指引的 Hugging Face 数据集下载 GR-1 评估数据放到datasets/目录含datasets/eval/video_131_cond_frame.png等条件帧和动作文件。2. 运行实时推理仓库提供了开箱即用的 launch_student_inference_teleop.sh等价于python -m cosmos_predict2._src.predict2.interactive.inference.action_video2world_teleop \ --configcosmos_predict2/_src/predict2/interactive/configs/config_distill.py \ --experimentcosmos_predict2p5_2B_action_gr00t_gr1_self_forcing_no_s3 \ --ckpt_path checkpoints/iter_000006000 \ --input_frame datasets/eval/video_131_cond_frame.png \ --fps 10.0 --num_steps 4 \ --action_source file --action_file datasets/eval/action_131.npy \ --max_latent_frames 100 \ --warmup --compile_model --save_output realtime.mp43. 关键参数速查参数含义实时化建议--num_steps每帧学生去噪步数保持 4步数越少越快--max_latent_frames最多生成多少 latent chunk100 约对应 400 帧像素40 秒 10FPS--warmup预热迭代先编译 CUDA kernel必加否则首帧很慢--compile_modeltorch.compile编译 DiT 与 VAE必加首次运行约 2 分钟编译--fast_mode一键开启全部优化想极致提速时使用--no_display无头模式服务器上用配合--save_output输出视频六、如何确认真的实时了脚本结束后会打印一份TIMING SUMMARY统计逻辑它会统计稳态下每个 chunk 的平均耗时毫秒并和播放预算每帧 100ms ÷ 帧数对比若出现✓ xx ms 1000 ms — REAL-TIME (1.06x)字样说明生成速度快于实时播放10 FPS 无压力本地显示窗口会用 OpenCV 实时渲染帧画面左上角叠加FPS: xx.x | Frame: N的实时帧率RealtimeVisualizer按q退出并保存结果。另外--skip_decode_display N可以让显示线程每 N 个 chunk 才解码一次把解码开销摊薄。七、常见问题排查 首次运行极慢正常现象torch.compile需要约 2 分钟做两次完整预热warmup 日志会提示之后进入稳态显存不足降低--max_latent_frames或使用--resolution 256,320降低分辨率LeapMotion 没有生效确认已替换占位实现为真实 SDK 数据否则日志中会出现 LeapMotion source not implemented 警告并回退随机动作更多问题参考官方排障文档 docs/ISSUES.md评测流程见 docs/EVAL.md。小结DreamDojo 的实时交互管线 蒸馏出的 4 步因果学生模型KV Cache 流式增量生成可插拔实时动作源。得益于无状态 KV Cache 与全链路编译单卡即可做到 10 FPS 稳定输出 1 分钟以上让看动作、立刻看到未来画面的机器人世界模型交互第一次成为现实。想继续深入可以从 docs/DISTILL.md 的蒸馏细节和 docs/POSTTRAIN.md 的后训练流程看起。赞分享【免费下载链接】DreamDojoOfficial Codebase for DreamDojo: A Generalist Robot World Model from Large-Scale Human Videos (ICML 2026)项目地址https://gitcode.com/gh_mirrors/dr/DreamDojo点击查看免费下载相关推荐LMCache KV Cache 迁移Move实战指南跨实例移动 KV Cache 的原理与操作LMCache KV Cache 迁移Move实战指南跨实例移动 KV Cache 的原理与操作 导读 本文聚焦 LMCache 提供的 move 接口人工智能大模型缓存抽象模型推理服务Gradio项目实战掌握流式输出功能实现实时交互体验Gradio项目实战掌握流式输出功能实现实时交互体验 什么是流式输出 在传统的人机交互应用中用户输入请求后需要等待系统完全处理完毕才能看到结果。而流式输出前端后端AI 应用MCP代理流式输出mcp-use实时交互开发MCP代理流式输出mcp use实时交互开发 MCP代理连接协议流式输出技术解决了传统批量响应模式下用户等待时间长、交互体验差的痛点。通过实时传输数据片段后端MCP 服务MCP ClientsAI Agent人工智能创作声明:本文部分内容由AI辅助生成(AIGC),仅供参考