$ show caiyuanhao1998/Open-OmniVCus
Open-OmniVCus is a research monorepo reproducing the NeurIPS 2025 OmniVCus method for subject-driven video customization under multimodal control conditions (reference image, depth, mask, motion/camera). It is hybrid by nature: the dominant character is an application/framework layer provided by a large vendored/forked copy of DiffSynth-Studio, wrapped around research-oriented data-construction and training code (VideoCus-Factory, Video-Depth-Anything). The repo does not expose a hosted service or API; it is a Python library plus executable training/inference scripts for diffusion-based image and video generation.
The LLM identified 4 files worth highlighting for reuse:
Single entry point for forward attention that selects flash-attn 2/3, SageAttention, xFormers, or PyTorch SDPA based on environment and availability. It is an isolated utility that can be dropped into any transformer implementation without importing the rest of the framework, and it demonstrates non-obvious backend fallback and precision handling.
Lazy mapping of safetensors to disk with on-demand tensor loading, enabling inference and training of multi-billion-parameter models on low-VRAM hardware. The pattern is reusable for any large-checkpoint loading problem and is substantially more sophisticated than a simple torch.load wrapper.
Composable data-loading DSL with operators chained via >> (e.g., ToAbsolutePath >> LoadImage >> ImageCropAndResize) and a UnifiedDataset that reads csv/json/jsonl metadata with repeat counts and modality routing. This is the cleanest reusable data-loading abstraction in the repository and can be adapted to other image/video/audio ML pipelines.
General LoRA loader/merger supporting multiple state-dict formats (internal, diffusers, civitai), alpha scaling, fusion, and rank reset. It is designed generically enough to be extracted and applied to other diffusion-model codebases with minimal modification.