diff --git a/modules/sdnq/common.py b/modules/sdnq/common.py index 698a35633..c5ff8ab9a 100644 --- a/modules/sdnq/common.py +++ b/modules/sdnq/common.py @@ -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"] diff --git a/modules/seedvr/config_7b.yaml b/modules/seedvr/config_7b.yaml index 61803415d..0e5cb146c 100644 --- a/modules/seedvr/config_7b.yaml +++ b/modules/seedvr/config_7b.yaml @@ -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