sarthak247 commited on
Commit
824e587
·
verified ·
1 Parent(s): bd75044

Add quantized diffusion model

Browse files
Files changed (2) hide show
  1. config.json +30 -0
  2. diffusion_pytorch_model.safetensors +3 -0
config.json ADDED
@@ -0,0 +1,30 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_class_name": "WanModel",
3
+ "_diffusers_version": "0.32.2",
4
+ "_name_or_path": "./Wan2.1-T2V-1.3B",
5
+ "dim": 1536,
6
+ "eps": 1e-06,
7
+ "ffn_dim": 8960,
8
+ "freq_dim": 256,
9
+ "in_dim": 16,
10
+ "model_type": "t2v",
11
+ "num_heads": 12,
12
+ "num_layers": 30,
13
+ "out_dim": 16,
14
+ "quantization_config": {
15
+ "_load_in_4bit": true,
16
+ "_load_in_8bit": false,
17
+ "bnb_4bit_compute_dtype": "float16",
18
+ "bnb_4bit_quant_storage": "uint8",
19
+ "bnb_4bit_quant_type": "nf4",
20
+ "bnb_4bit_use_double_quant": true,
21
+ "llm_int8_enable_fp32_cpu_offload": false,
22
+ "llm_int8_has_fp16_weight": false,
23
+ "llm_int8_skip_modules": null,
24
+ "llm_int8_threshold": 6.0,
25
+ "load_in_4bit": true,
26
+ "load_in_8bit": false,
27
+ "quant_method": "bitsandbytes"
28
+ },
29
+ "text_len": 512
30
+ }
diffusion_pytorch_model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:91bede864e608c914d447d11a7bc26b1b54824e6a4203ce62aeef217da157d3b
3
+ size 734657595