|
{
|
|
"_class_name": "HunyuanVideoTransformer3DModel",
|
|
"_diffusers_version": "0.33.0.dev0",
|
|
"_name_or_path": "Skywork/SkyReels-V1-Hunyuan-I2V",
|
|
"attention_head_dim": 128,
|
|
"guidance_embeds": true,
|
|
"in_channels": 32,
|
|
"mlp_ratio": 4.0,
|
|
"num_attention_heads": 24,
|
|
"num_layers": 20,
|
|
"num_refiner_layers": 2,
|
|
"num_single_layers": 40,
|
|
"out_channels": 16,
|
|
"patch_size": 2,
|
|
"patch_size_t": 1,
|
|
"pooled_projection_dim": 768,
|
|
"qk_norm": "rms_norm",
|
|
"quantization_config": {
|
|
"_load_in_4bit": true,
|
|
"_load_in_8bit": false,
|
|
"bnb_4bit_compute_dtype": "bfloat16",
|
|
"bnb_4bit_quant_storage": "uint8",
|
|
"bnb_4bit_quant_type": "nf4",
|
|
"bnb_4bit_use_double_quant": false,
|
|
"llm_int8_enable_fp32_cpu_offload": false,
|
|
"llm_int8_has_fp16_weight": false,
|
|
"llm_int8_skip_modules": null,
|
|
"llm_int8_threshold": 6.0,
|
|
"load_in_4bit": true,
|
|
"load_in_8bit": false,
|
|
"quant_method": "bitsandbytes"
|
|
},
|
|
"rope_axes_dim": [
|
|
16,
|
|
56,
|
|
56
|
|
],
|
|
"rope_theta": 256.0,
|
|
"text_embed_dim": 4096
|
|
}
|
|
|