v1 fix bad vae
Browse files- vae/config.json +0 -47
vae/config.json
CHANGED
|
@@ -1,7 +1,6 @@
|
|
| 1 |
{
|
| 2 |
"_class_name": "AutoencoderKLQwenImage",
|
| 3 |
"_diffusers_version": "0.35.0.dev0",
|
| 4 |
-
"_name_or_path": "./tmp_model/vae",
|
| 5 |
"attn_scales": [],
|
| 6 |
"base_dim": 96,
|
| 7 |
"dim_mult": [
|
|
@@ -48,52 +47,6 @@
|
|
| 48 |
1.916
|
| 49 |
],
|
| 50 |
"num_res_blocks": 2,
|
| 51 |
-
"quantization_config": {
|
| 52 |
-
"_load_in_4bit": true,
|
| 53 |
-
"_load_in_8bit": false,
|
| 54 |
-
"bnb_4bit_compute_dtype": "bfloat16",
|
| 55 |
-
"bnb_4bit_quant_storage": "uint8",
|
| 56 |
-
"bnb_4bit_quant_type": "nf4",
|
| 57 |
-
"bnb_4bit_use_double_quant": true,
|
| 58 |
-
"llm_int8_enable_fp32_cpu_offload": false,
|
| 59 |
-
"llm_int8_has_fp16_weight": false,
|
| 60 |
-
"llm_int8_skip_modules": [
|
| 61 |
-
"transformer_blocks.0.img_mod.1",
|
| 62 |
-
"transformer_blocks.0.attn.to_q",
|
| 63 |
-
"transformer_blocks.0.attn.to_k",
|
| 64 |
-
"transformer_blocks.0.attn.to_v",
|
| 65 |
-
"transformer_blocks.0.attn.add_k_proj",
|
| 66 |
-
"transformer_blocks.0.attn.add_v_proj",
|
| 67 |
-
"transformer_blocks.0.attn.add_q_proj",
|
| 68 |
-
"transformer_blocks.0.attn.to_out.0",
|
| 69 |
-
"transformer_blocks.0.attn.to_add_out",
|
| 70 |
-
"transformer_blocks.0.img_mlp.net.0.proj",
|
| 71 |
-
"transformer_blocks.0.img_mlp.net.2",
|
| 72 |
-
"transformer_blocks.0.txt_mod.1",
|
| 73 |
-
"transformer_blocks.0.txt_mlp.net.0.proj",
|
| 74 |
-
"transformer_blocks.0.txt_mlp.net.2",
|
| 75 |
-
"transformer_blocks.59.img_mod.1",
|
| 76 |
-
"transformer_blocks.59.attn.to_q",
|
| 77 |
-
"transformer_blocks.59.attn.to_k",
|
| 78 |
-
"transformer_blocks.59.attn.to_v",
|
| 79 |
-
"transformer_blocks.59.attn.add_k_proj",
|
| 80 |
-
"transformer_blocks.59.attn.add_v_proj",
|
| 81 |
-
"transformer_blocks.59.attn.add_q_proj",
|
| 82 |
-
"transformer_blocks.59.attn.to_out.0",
|
| 83 |
-
"transformer_blocks.59.attn.to_add_out",
|
| 84 |
-
"transformer_blocks.59.img_mlp.net.0.proj",
|
| 85 |
-
"transformer_blocks.59.img_mlp.net.2",
|
| 86 |
-
"transformer_blocks.59.txt_mod.1",
|
| 87 |
-
"transformer_blocks.59.txt_mlp.net.0.proj",
|
| 88 |
-
"transformer_blocks.59.txt_mlp.net.2",
|
| 89 |
-
"norm_out.linear",
|
| 90 |
-
"proj_out"
|
| 91 |
-
],
|
| 92 |
-
"llm_int8_threshold": 6.0,
|
| 93 |
-
"load_in_4bit": true,
|
| 94 |
-
"load_in_8bit": false,
|
| 95 |
-
"quant_method": "bitsandbytes"
|
| 96 |
-
},
|
| 97 |
"temperal_downsample": [
|
| 98 |
false,
|
| 99 |
true,
|
|
|
|
| 1 |
{
|
| 2 |
"_class_name": "AutoencoderKLQwenImage",
|
| 3 |
"_diffusers_version": "0.35.0.dev0",
|
|
|
|
| 4 |
"attn_scales": [],
|
| 5 |
"base_dim": 96,
|
| 6 |
"dim_mult": [
|
|
|
|
| 47 |
1.916
|
| 48 |
],
|
| 49 |
"num_res_blocks": 2,
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 50 |
"temperal_downsample": [
|
| 51 |
false,
|
| 52 |
true,
|