SDNQ add NaDiT keys

This commit is contained in:
Disty0
2025-10-14 17:18:58 +03:00
parent 57230bbf3a
commit 4aee524ddf
2 changed files with 7 additions and 1 deletions
+6
View File
@@ -95,4 +95,10 @@ module_skip_keys_dict = {
["transformer_blocks.0.img_mod.1.weight", ".time_text_embed", ".txt_in", ".img_in", ".proj_out", ".norm_out", "pos_embed"],
{}
],
"NaDiT": [
[".emb_in", ".txt_in", ".vid_in", ".emb_scale", ".vid_out", ".vid_out_norm", ".vid_out_ada"],
{}
],
}
module_skip_keys_dict["NaDiTUpscaler"] = module_skip_keys_dict["NaDiT"]
+1 -1
View File
@@ -8,7 +8,7 @@ dit:
path:
- "SeedVR2_VideoUpscaler.src.models.dit.nadit"
- "SeedVR2_VideoUpscaler.src.models.dit.nadit"
- "src.models.dit.nadit"
- "modules.seedvr.src.models.dit.nadit"
name: "NaDiT"
args: "as_params"
vid_in_channels: 33