Wan2.1-T2V-1.3B-nf4 / config.json
sarthak247's picture
Add quantized diffusion model
824e587 verified
raw
history blame contribute delete
769 Bytes
{
"_class_name": "WanModel",
"_diffusers_version": "0.32.2",
"_name_or_path": "./Wan2.1-T2V-1.3B",
"dim": 1536,
"eps": 1e-06,
"ffn_dim": 8960,
"freq_dim": 256,
"in_dim": 16,
"model_type": "t2v",
"num_heads": 12,
"num_layers": 30,
"out_dim": 16,
"quantization_config": {
"_load_in_4bit": true,
"_load_in_8bit": false,
"bnb_4bit_compute_dtype": "float16",
"bnb_4bit_quant_storage": "uint8",
"bnb_4bit_quant_type": "nf4",
"bnb_4bit_use_double_quant": true,
"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"
},
"text_len": 512
}