o vi@sLdZddlmZddlmZddlmZeedddddd d Zd d Z d S) zJ Cosmos Policy tokenizer registration with deterministic seeding support. ) ConfigStore)LazyCall)Wan2pt1VAEInterfacezBhf://nvidia/Cosmos-Predict2-2B-Video2World/tokenizer/tokenizer.pthzcredentials/s3_training.secretFN)vae_pths3_credential_path load_mean_stdtemporal_window is_parallel cp_grid_shapecCst}|jdddtddS)z Register Cosmos Policy tokenizer configurations. This registers the wan2pt1 tokenizer with deterministic seeding support. To enable deterministic encoding, set: DETERMINISTIC=true tokenizerzmodel.config.tokenizerpolicy_wan2pt1_tokenizer)grouppackagenamenodeN)rinstancestorePolicyWan2pt1VAEConfig)csrM/data/cameron/vidgen/cosmos-policy/cosmos_policy/config/defaults/tokenizer.pyregister_policy_tokenizer$s r) __doc__hydra.core.config_storer)cosmos_policy._src.imaginaire.lazy_configrLZ cosmos_policy.tokenizers.wan2pt1rrrrrrrs