vllm_omni.diffusion.models.internvla_a1.model_cosmos ¶ ImageTokenizer ¶ Bases: Module decode ¶ decode(input_latent: Tensor) -> Tensor encode ¶ encode(input_tensor: Tensor) -> Tensor infer_cosmos_ci_spatial_compression ¶ infer_cosmos_ci_spatial_compression( path: str | Path, ) -> int is_safetensors_checkpoint ¶ is_safetensors_checkpoint(path: str | Path) -> bool load_cosmos_component ¶ load_cosmos_component( checkpoint_filepath: str | Path, *, component: str, device: str = "cuda", ) -> Module