File size: 1,752 Bytes
6092fb8 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 | {
"_class_name": "CosmosTransformer3DModel",
"_diffusers_version": "0.38.0",
"adaln_lora_dim": 256,
"attention_head_dim": 128,
"concat_padding_mask": true,
"controlnet_block_every_n": null,
"crossattn_proj_in_channels": 1024,
"encoder_hidden_states_channels": 1024,
"extra_pos_embed_type": null,
"img_context_dim_in": null,
"img_context_dim_out": 2048,
"img_context_num_tokens": 256,
"in_channels": 16,
"max_size": [
128,
240,
240
],
"mlp_ratio": 4.0,
"num_attention_heads": 16,
"num_layers": 28,
"out_channels": 16,
"patch_size": [
1,
2,
2
],
"quantization_config": {
"add_skip_keys": false,
"dequantize_fp32": false,
"dynamic_loss_threshold": null,
"group_size": 0,
"is_integer": true,
"is_training": false,
"modules_dtype_dict": {},
"modules_quant_config": {},
"modules_to_not_convert": [
"crossattn_proj",
"patch_embed",
"transformer_blocks.0.norm*",
"norm_out",
"proj_out",
"learnable_pos_embed",
"time_embed"
],
"non_blocking": false,
"quant_conv": false,
"quant_embedding": false,
"quant_method": "sdnq",
"quantization_device": null,
"quantized_matmul_dtype": null,
"return_device": null,
"sdnq_version": "0.1.8",
"svd_rank": 32,
"svd_steps": 8,
"use_dynamic_quantization": false,
"use_grad_ckpt": true,
"use_quantized_matmul": true,
"use_quantized_matmul_conv": false,
"use_static_quantization": true,
"use_stochastic_rounding": false,
"use_svd": false,
"weights_dtype": "uint4"
},
"rope_scale": [
1.0,
4.0,
4.0
],
"text_embed_dim": 1024,
"use_crossattn_projection": false,
"_name_or_path": null
}
|