Cosmos
Safetensors
NeMo
cosmos-embed1
nvidia
custom_code
Cosmos-Embed1-448p / config.json
fferroni's picture
First commit
f653621
{
"architectures": [
"CosmosEmbed1"
],
"auto_map": {
"AutoConfig": "configuration_embed1.CosmosEmbed1Config",
"AutoProcessor": "preprocessing_embed1.CosmosEmbed1Processor",
"AutoModel": "modeling_embed1.CosmosEmbed1"
},
"embed_dim": 768,
"max_txt_len": 128,
"model_type": "cosmos-embed1",
"num_query_tokens": 32,
"num_video_frames": 8,
"resolution": 448,
"temporal_encoding_type": "neighboring_token_propagation",
"transformers_version": "4.51.3",
"vocab_size": 30523,
"transformer_engine": false,
"use_fp8": false
}