amap-cvlab/ABot-World
Python0 / 0
단일 RTX 5090에서 실시간 무한 상호작용 월드 생성
TL;DR
이 레포지토리는 단일 데스크탑 GPU에서 실시간 상호작용 월드를 생성하는 인퍼런스 코드, 미리 학습된 모델과 대화형 데모를 포함하는 standalone 도구입니다. HuggingFace·ModelScope에 모델과 데이터셋이 호스팅되어 있고 Reactor 웹 플레이그라운드와 로컬 Gradio 데모로 즉시 체험할 수 있습니다. 성능 목표는 RTX 5090·CUDA 12.8 환경에서 720p·16FPS, 1.2초 지연, 19GB 메모리이며 README에 상세한 네이티브 빌드·설치 절차가 수록되어 있습니다. 설치와 네이티브 커널 빌드 복잡성을 감수할 수 있고 데스크탑 기반의 오픈엔디드 상호작용 월드가 필요하다면 검토할 가치가 충분합니다.
핵심 포인트
- ABot-World는 사용자의 행동 입력을 받아 즉시 반응하는 action-conditioned world rollout 기능을 제공합니다. 내부적으로는 행동을 트리거로 연속 프레임을 생성하는 인퍼런스 파이프라인을 causal student diffusion generator와 LongForcing 훈련 전략으로 연결해 다음 장면을 예측하고 확장합니다. 이 방식은 고정 길이 비디오 재생이 아닌 계속되는 탐색을 가능하게 하여 인터랙티브 애플리케이션에서 사용자가 환경을 실시간으로 조작할 수 있게 합니다.
- 실시간 데스크탑 추론 성능을 위해 FlashAttention, SageAttention, LightX2V, Triton 최적화 커널 등 하드웨어 특화 연산을 통합하고 네이티브 빌드 절차를 적용합니다. 설치 단계에서는 CUDA 12.8 환경과 특정 wheel 바이너리, 소스 빌드가 요구되며 README에 각 빌드 명령과 의존성 경로가 상세히 포함되어 있습니다. 이 최적화 파이프라인은 단일 NVIDIA RTX 5090에서 19GB 메모리로 720p·16FPS 수준의 실시간 응답을 달성하도록 설계되어 데스크탑 실험에 적합한 성능-효율 균형을 제공합니다.
- 무한 롤아웃 개념은 LongForcing 훈련으로 장면 고착(scene lock-in)을 회피하고 모델이 새로운 장면과 동적 변화를 계속 생성하도록 유도합니다. 학습 파이프라인은 ABot-World-0-5B-LF causal student 모델과 500시간 분량의 행동 주석 비디오 데이터셋을 활용해 연속성과 다양성을 유지하면서 장면을 확장하는 방식으로 동작합니다. 이 접근은 고정 길이 비디오 한계를 넘어서서 오픈엔디드 상호작용 월드를 필요로 하는 연구·프로토타입에 직접적인 이점을 제공합니다.
- 프로젝트는 여러 네이티브 종속성과 커널 빌드를 요구하므로 환경 구성 난이도가 높습니다. FlashAttention wheel 호환성, SageAttention 소스 설치, LightX2V의 CUTLASS 경로 설정 등 운영체제·CUDA·glibc 버전 차이에서 발생하는 이슈를 직접 해결해야 할 가능성이 큽니다. 따라서 빠른 시범 운용을 원한다면 고사양 하드웨어와 GPU 네이티브 빌드 경험이 필요하며, 이러한 설치 비용이 도입 결정의 핵심 고려사항이 됩니다.
벤치마크
| 벤치마크 | 지표 | 값 | 비교 |
|---|---|---|---|
| Inference performance | resolution·FPS·latency·GPU memory | 720p, 16 FPS, 1.2s latency, 19GB | — |
| Model | causal student checkpoint | ABot-World-0-5B-LF (distilled generator weights merged into diffusion_pytorch_model.safetensors) | — |
| Dataset | training data size | 500-hour action-annotated video dataset | — |
이미지 분석


1.7k
Stars
39
Forks
+208
Trending
0
조회수
1.7k watchers3 open issues
관련 토론
아직 관련 토론이 없습니다.
댓글
댓글을 작성하려면 로그인이 필요합니다.