请求适配新模型 BS-Roformer类型 #101
Description
整合包使用版本为：MSST_WebUI_1.7.0_v2_cu128.7z
运行环境：Windows 11 & CUDA
模型链接为：pcunwa/BS-Roformer-HyperACE
使用导入非官方模型功能后，进行推理，终端报错如下：
21:05:48.093 [ERROR] [webui\msst.py:206] Separation failed: Error(s) in loading state_dict for BSRoformer:
Unexpected key(s) in state_dict: "mask_estimators.0.segm.backbone.stem.dwconv.weight", "mask_estimators.0.segm.backbone.stem.pwconv.weight", "mask_estimators.0.segm.backbone.stem.bn.weight", "mask_estimators.0.segm.backbone.stem.bn.bias", "mask_estimators.0.segm.backbone.p2.0.dwconv.weight", "mask_estimators.0.segm.backbone.p2.0.pwconv.weight", "mask_estimators.0.segm.backbone.p2.0.bn.weight", "mask_estimators.0.segm.backbone.p2.0.bn.bias", "mask_estimators.0.segm.backbone.p2.1.cv1.conv.weight", "mask_estimators.0.segm.backbone.p2.1.cv1.bn.weight", "mask_estimators.0.segm.backbone.p2.1.cv1.bn.bias", "mask_estimators.0.segm.backbone.p2.1.m.cv1.conv.weight", "mask_estimators.0.segm.backbone.p2.1.m.cv1.bn.weight", "mask_estimators.0.segm.backbone.p2.1.m.cv1.bn.bias", "mask_estimators.0.segm.backbone.p2.1.m.cv2.conv.weight", "mask_estimators.0.segm.backbone.p2.1.m.cv2.bn.weight", "mask_estimators.0.segm.backbone.p2.1.m.cv2.bn.bias", "mask_estimators.0.segm.backbone.p2.1.m.cv3.conv.weight", "mask_estimators.0.segm.backbone.p2.1.m.cv3.bn.weight", "mask_estimators.0.segm.backbone.p2.1.m.cv3.bn.bias", "mask_estimators.0.segm.backbone.p2.1.m.m.0.dsconv1.dwconv.weight", "mask_estimators.0.segm.backbone.p2.1.m.m.0.dsconv1.pwconv.weight", "mask_estimators.0.segm.backbone.p2.1.m.m.0.dsconv1.bn.weight", "mask_estimators.0.segm.backbone.p2.1.m.m.0.dsconv1.bn.bias", "mask_estimators.0.segm.backbone.p2.1.m.m.0.dsconv2.dwconv.weight", "mask_estimators.0.segm.backbone.p2.1.m.m.0.dsconv2.pwconv.weight", "mask_estimators.0.segm.backbone.p2.1.m.m.0.dsconv2.bn.weight", "mask_estimators.0.segm.backbone.p2.1.m.m.0.dsconv2.bn.bias", "mask_estimators.0.segm.backbone.p2.1.m.m.1.dsconv1.dwconv.weight", "mask_estimators.0.segm.backbone.p2.1.m.m.1.dsconv1.pwconv.weight", "mask_estimators.0.segm.backbone.p2.1.m.m.1.dsconv1.bn.weight", "mask_estimators.0.segm.backbone.p2.1.m.m.1.dsconv1.bn.bias", "mask_estimators.0.segm.backbone.p2.1.m.m.1.dsconv2.dwconv.weight", "mask_estimators.0.segm.backbone.p2.1.m.m.1.dsconv2.pwconv.weight", "mask_estimators.0.segm.backbone.p2.1.m.m.1.dsconv2.bn.weight", "mask_estimators.0.segm.backbone.p2.1.m.m.1.dsconv2.bn.bias", "mask_estimators.0.segm.backbone.p2.1.cv2.conv.weight", "mask_estimators.0.segm.backbone.p2.1.cv2.bn.weight", "mask_estimators.0.segm.backbone.p2.1.cv2.bn.bias", "mask_estimators.0.segm.backbone.p3.0.dwconv.weight", "mask_estimators.0.segm.backbone.p3.0.pwconv.weight", "mask_estimators.0.segm.backbone.p3.0.bn.weight", "mask_estimators.0.segm.backbone.p3.0.bn.bias", "mask_estimators.0.segm.backbone.p3.1.cv1.conv.weight", "mask_estimators.0.segm.backbone.p3.1.cv1.bn.weight", "mask_estimators.0.segm.backbone.p3.1.cv1.bn.bias", "mask_estimators.0.segm.backbone.p3.1.m.cv1.conv.weight", "mask_estimators.0.segm.backbone.p3.1.m.cv1.bn.weight", "mask_estimators.0.segm.backbone.p3.1.m.cv1.bn.bias", "mask_estimators.0.segm.backbone.p3.1.m.cv2.conv.weight", "mask_estimators.0.segm.backbone.p3.1.m.cv2.bn.weight", "mask_estimators.0.segm.backbone.p3.1.m.cv2.bn.bias", "mask_estimators.0.segm.backbone.p3.1.m.cv3.conv.weight", "mask_estimators.0.segm.backbone.p3.1.m.cv3.bn.weight", "mask_estimators.0.segm.backbone.p3.1.m.cv3.bn.bias", "mask_estimators.0.segm.backbone.p3.1.m.m.0.dsconv1.dwconv.weight", "mask_estimators.0.segm.backbone.p3.1.m.m.0.dsconv1.pwconv.weight", "mask_estimators.0.segm.backbone.p3.1.m.m.0.dsconv1.bn.weight", "mask_estimators.0.segm.backbone.p3.1.m.m.0.dsconv1.bn.bias", "mask_estimators.0.segm.backbone.p3.1.m.m.0.dsconv2.dwconv.weight", "mask_estimators.0.segm.backbone.p3.1.m.m.0.dsconv2.pwconv.weight", "mask_estimators.0.segm.backbone.p3.1.m.m.0.dsconv2.bn.weight", "mask_estimators.0.segm.backbone.p3.1.m.m.0.dsconv2.bn.bias", "mask_estimators.0.segm.backbone.p3.1.m.m.1.dsconv1.dwconv.weight", "mask_estimators.0.segm.backbone.p3.1.m.m.1.dsconv1.pwconv.weight", "mask_estimators.0.segm.backbone.p3.1.m.m.1.dsconv1.bn.weight", "mask_estimators.0.segm.backbone.p3.1.m.m.1.dsconv1.bn.bias", "mask_estimators.0.segm.backbone.p3.1.m.m.1.dsconv2.dwconv.weight", "mask_estimators.0.segm.backbone.p3.1.m.m.1.dsconv2.pwconv.weight", "mask_estimators.0.segm.backbone.p3.1.m.m.1.dsconv2.bn.weight", "mask_estimators.0.segm.backbone.p3.1.m.m.1.dsconv2.bn.bias", "mask_estimators.0.segm.backbone.p3.1.m.m.2.dsconv1.dwconv.weight", "mask_estimators.0.segm.backbone.p3.1.m.m.2.dsconv1.pwconv.weight", "mask_estimators.0.segm.backbone.p3.1.m.m.2.dsconv1.bn.weight", "mask_estimators.0.segm.backbone.p3.1.m.m.2.dsconv1.bn.bias", "mask_estimators.0.segm.backbone.p3.1.m.m.2.dsconv2.dwconv.weight", "mask_estimators.0.segm.backbone.p3.1.m.m.2.dsconv2.pwconv.weight", "mask_estimators.0.segm.backbone.p3.1.m.m.2.dsconv2.bn.weight", "mask_estimators.0.segm.backbone.p3.1.m.m.2.dsconv2.bn.bias", "mask_estimators.0.segm.backbone.p3.1.m.m.3.dsconv1.dwconv.weight", "mask_estimators.0.segm.backbone.p3.1.m.m.3.dsconv1.pwconv.weight", "mask_estimators.0.segm.backbone.p3.1.m.m.3.dsconv1.bn.weight", "mask_estimators.0.segm.backbone.p3.1.m.m.3.dsconv1.bn.bias", "mask_estimators.0.segm.backbone.p3.1.m.m.3.dsconv2.dwconv.weight", "mask_estimators.0.segm.backbone.p3.1.m.m.3.dsconv2.pwconv.weight", "mask_estimators.0.segm.backbone.p3.1.m.m.3.dsconv2.bn.weight", "mask_estimators.0.segm.backbone.p3.1.m.m.3.dsconv2.bn.bias", "mask_estimators.0.segm.backbone.p3.1.cv2.conv.weight", "mask_estimators.0.segm.backbone.p3.1.cv2.bn.weight", "mask_estimators.0.segm.backbone.p3.1.cv2.bn.bias", "mask_estimators.0.segm.backbone.p4.0.dwconv.weight", "mask_estimators.0.segm.backbone.p4.0.pwconv.weight", "mask_estimators.0.segm.backbone.p4.0.bn.weight", "mask_estimators.0.segm.backbone.p4.0.bn.bias", "mask_estimators.0.segm.backbone.p4.1.cv1.conv.weight", "mask_estimators.0.segm.backbone.p4.1.cv1.bn.weight", "mask_estimators.0.segm.backbone.p4.1.cv1.bn.bias", "mask_estimators.0.segm.backbone.p4.1.m.cv1.conv.weight", "mask_estimators.0.segm.backbone.p4.1.m.cv1.bn.weight", "mask_estimators.0.segm.backbone.p4.1.m.cv1.bn.bias", "mask_estimators.0.segm.backbone.p4.1.m.cv2.conv.weight", "mask_estimators.0.segm.backbone.p4.1.m.cv2.bn.weight", "mask_estimators.0.segm.backbone.p4.1.m.cv2.bn.bias", "mask_estimators.0.segm.backbone.p4.1.m.cv3.conv.weight", "mask_estimators.0.segm.backbone.p4.1.m.cv3.bn.weight", "mask_estimators.0.segm.backbone.p4.1.m.cv3.bn.bias", "mask_estimators.0.segm.backbone.p4.1.m.m.0.dsconv1.dwconv.weight", "mask_estimators.0.segm.backbone.p4.1.m.m.0.dsconv1.pwconv.weight", "mask_estimators.0.segm.backbone.p4.1.m.m.0.dsconv1.bn.weight", "mask_estimators.0.segm.backbone.p4.1.m.m.0.dsconv1.bn.bias", "mask_estimators.0.segm.backbone.p4.1.m.m.0.dsconv2.dwconv.weight", "mask_estimators.0.segm.backbone.p4.1.m.m.0.dsconv2.pwconv.weight", "mask_estimators.0.segm.backbone.p4.1.m.m.0.dsconv2.bn.weight", "mask_estimators.0.segm.backbone.p4.1.m.m.0.dsconv2.bn.bias", "mask_estimators.0.segm.backbone.p4.1.m.m.1.dsconv1.dwconv.weight", "mask_estimators.0.segm.backbone.p4.1.m.m.1.dsconv1.pwconv.weight", "mask_estimators.0.segm.backbone.p4.1.m.m.1.dsconv1.bn.weight", "mask_estimators.0.segm.backbone.p4.1.m.m.1.dsconv1.bn.bias", "mask_estimators.0.segm.backbone.p4.1.m.m.1.dsconv2.dwconv.weight", "mask_estimators.0.segm.backbone.p4.1.m.m.1.dsconv2.pwconv.weight", "mask_estimators.0.segm.backbone.p4.1.m.m.1.dsconv2.bn.weight", "mask_estimators.0.segm.backbone.p4.1.m.m.1.dsconv2.bn.bias", "mask_estimators.0.segm.backbone.p4.1.m.m.2.dsconv1.dwconv.weight", "mask_estimators.0.segm.backbone.p4.1.m.m.2.dsconv1.pwconv.weight", "mask_estimators.0.segm.backbone.p4.1.m.m.2.dsconv1.bn.weight", "mask_estimators.0.segm.backbone.p4.1.m.m.2.dsconv1.bn.bias", "mask_estimators.0.segm.backbone.p4.1.m.m.2.dsconv2.dwconv.weight", "mask_estimators.0.segm.backbone.p4.1.m.m.2.dsconv2.pwconv.weight", "mask_estimators.0.segm.backbone.p4.1.m.m.2.dsconv2.bn.weight", "mask_estimators.0.segm.backbone.p4.1.m.m.2.dsconv2.bn.bias", "mask_estimators.0.segm.backbone.p4.1.m.m.3.dsconv1.dwconv.weight", "mask_estimators.0.segm.backbone.p4.1.m.m.3.dsconv1.pwconv.weight", "mask_estimators.0.segm.backbone.p4.1.m.m.3.dsconv1.bn.weight", "mask_estimators.0.segm.backbone.p4.1.m.m.3.dsconv1.bn.bias", "mask_estimators.0.segm.backbone.p4.1.m.m.3.dsconv2.dwconv.weight", "mask_estimators.0.segm.backbone.p4.1.m.m.3.dsconv2.pwconv.weight", "mask_estimators.0.segm.backbone.p4.1.m.m.3.dsconv2.bn.weight", "mask_estimators.0.segm.backbone.p4.1.m.m.3.dsconv2.bn.bias", "mask_estimators.0.segm.backbone.p4.1.cv2.conv.weight", "mask_estimators.0.segm.backbone.p4.1.cv2.bn.weight", "mask_estimators.0.segm.backbone.p4.1.cv2.bn.bias", "mask_estimators.0.segm.backbone.p5.0.dwconv.weight", "mask_estimators.0.segm.backbone.p5.0.pwconv.weight", "mask_estimators.0.segm.backbone.p5.0.bn.weight", "mask_estimators.0.segm.backbone.p5.0.bn.bias", "mask_estimators.0.segm.backbone.p5.1.cv1.conv.weight", "mask_estimators.0.segm.backbone.p5.1.cv1.bn.weight", "mask_estimators.0.segm.backbone.p5.1.cv1.bn.bias", "mask_estimators.0.segm.backbone.p5.1.m.cv1.conv.weight", "mask_estimators.0.segm.backbone.p5.1.m.cv1.bn.weight", "mask_estimators.0.segm.backbone.p5.1.m.cv1.bn.bias", "mask_estimators.0.segm.backbone.p5.1.m.cv2.conv.weight", "mask_estimators.0.segm.backbone.p5.1.m.cv2.bn.weight", "mask_estimators.0.segm.backbone.p5.1.m.cv2.bn.bias", "mask_estimators.0.segm.backbone.p5.1.m.cv3.conv.weight", "mask_estimators.0.segm.backbone.p5.1.m.cv3.bn.weight", "mask_estimators.0.segm.backbone.p5.1.m.cv3.bn.bias", "mask_estimators.0.segm.backbone.p5.1.m.m.0.dsconv1.dwconv.weight", "mask_estimators.0.segm.backbone.p5.1.m.m.0.dsconv1.pwconv.weight", "mask_estimators.0.segm.backbone.p5.1.m.m.0.dsconv1.bn.weight", "mask_estimators.0.segm.backbone.p5.1.m.m.0.dsconv1.bn.bias", "mask_estimators.0.segm.backbone.p5.1.m.m.0.dsconv2.dwconv.weight", "mask_estimators.0.segm.backbone.p5.1.m.m.0.dsconv2.pwconv.weight", "mask_estimators.0.segm.backbone.p5.1.m.m.0.dsconv2.bn.weight", "mask_estimators.0.segm.backbone.p5.1.m.m.0.dsconv2.bn.bias", "mask_estimators.0.segm.backbone.p5.1.m.m.1.dsconv1.dwconv.weight", "mask_estimators.0.segm.backbone.p5.1.m.m.1.dsconv1.pwconv.weight", "mask_estimators.0.segm.backbone.p5.1.m.m.1.dsconv1.bn.weight", "mask_estimators.0.segm.backbone.p5.1.m.m.1.dsconv1.bn.bias", "mask_estimators.0.segm.backbone.p5.1.m.m.1.dsconv2.dwconv.weight", "mask_estimators.0.segm.backbone.p5.1.m.m.1.dsconv2.pwconv.weight", "mask_estimators.0.segm.backbone.p5.1.m.m.1.dsconv2.bn.weight", "mask_estimators.0.segm.backbone.p5.1.m.m.1.dsconv2.bn.bias", "mask_estimators.0.segm.backbone.p5.1.cv2.conv.weight", "mask_estimators.0.segm.backbone.p5.1.cv2.bn.weight", "mask_estimators.0.segm.backbone.p5.1.cv2.bn.bias", "mask_estimators.0.segm.hyperace.fuse_conv.conv.weight", "mask_estimators.0.segm.hyperace.fuse_conv.bn.weight", "mask_estimators.0.segm.hyperace.fuse_conv.bn.bias", "mask_estimators.0.segm.hyperace.high_order_branch.0.cv1.conv.weight", "mask_estimators.0.segm.hyperace.high_order_branch.0.cv1.bn.weight", "mask_estimators.0.segm.hyperace.high_order_branch.0.cv1.bn.bias", "mask_estimators.0.segm.hyperace.high_order_branch.0.cv2.conv.weight", "mask_estimators.0.segm.hyperace.high_order_branch.0.cv2.bn.weight", "mask_estimators.0.segm.hyperace.high_order_branch.0.cv2.bn.bias", "mask_estimators.0.segm.hyperace.high_order_branch.0.ahc.adaptive_hyperedge_gen.global_proto", "mask_estimators.0.segm.hyperace.high_order_branch.0.ahc.adaptive_hyperedge_gen.context_mapper.weight", "mask_estimators.0.segm.hyperace.high_order_branch.0.ahc.adaptive_hyperedge_gen.query_proj.weight", "mask_estimators.0.segm.hyperace.high_order_branch.0.ahc.hypergraph_conv.W_e.weight", "mask_estimators.0.segm.hyperace.high_order_branch.0.ahc.hypergraph_conv.W_v.weight", "mask_estimators.0.segm.hyperace.high_order_branch.0.cv3.conv.weight", "mask_estimators.0.segm.hyperace.high_order_branch.0.cv3.bn.weight", "mask_estimators.0.segm.hyperace.high_order_branch.0.cv3.bn.bias", "mask_estimators.0.segm.hyperace.high_order_branch.1.cv1.conv.weight", "mask_estimators.0.segm.hyperace.high_order_branch.1.cv1.bn.weight", "mask_estimators.0.segm.hyperace.high_order_branch.1.cv1.bn.bias", "mask_estimators.0.segm.hyperace.high_order_branch.1.cv2.conv.weight", "mask_estimators.0.segm.hyperace.high_order_branch.1.cv2.bn.weight", "mask_estimators.0.segm.hyperace.high_order_branch.1.cv2.bn.bias", "mask_estimators.0.segm.hyperace.high_order_branch.1.ahc.adaptive_hyperedge_gen.global_proto", "mask_estimators.0.segm.hyperace.high_order_branch.1.ahc.adaptive_hyperedge_gen.context_mapper.weight", "mask_estimators.0.segm.hyperace.high_order_branch.1.ahc.adaptive_hyperedge_gen.query_proj.weight", "mask_estimators.0.segm.hyperace.high_order_branch.1.ahc.hypergraph_conv.W_e.weight", "mask_estimators.0.segm.hyperace.high_order_branch.1.ahc.hypergraph_conv.W_v.weight", "mask_estimators.0.segm.hyperace.high_order_branch.1.cv3.conv.weight", "mask_estimators.0.segm.hyperace.high_order_branch.1.cv3.bn.weight", "mask_estimators.0.segm.hyperace.high_order_branch.1.cv3.bn.bias", "mask_estimators.0.segm.hyperace.high_order_branch.2.cv1.conv.weight", "mask_estimators.0.segm.hyperace.high_order_branch.2.cv1.bn.weight", "mask_estimators.0.segm.hyperace.high_order_branch.2.cv1.bn.bias", "mask_estimators.0.segm.hyperace.high_order_branch.2.cv2.conv.weight", "mask_estimators.0.segm.hyperace.high_order_branch.2.cv2.bn.weight", "mask_estimators.0.segm.hyperace.high_order_branch.2.cv2.bn.bias", "mask_estimators.0.segm.hyperace.high_order_branch.2.ahc.adaptive_hyperedge_gen.global_proto", "mask_estimators.0.segm.hyperace.high_order_branch.2.ahc.adaptive_hyperedge_gen.context_mapper.weight", "mask_estimators.0.segm.hyperace.high_order_branch.2.ahc.adaptive_hyperedge_gen.query_proj.weight", "mask_estimators.0.segm.hyperace.high_order_branch.2.ahc.hypergraph_conv.W_e.weight", "mask_estimators.0.segm.hyperace.high_order_branch.2.ahc.hypergraph_conv.W_v.weight", "mask_estimators.0.segm.hyperace.high_order_branch.2.cv3.conv.weight", "mask_estimators.0.segm.hyperace.high_order_branch.2.cv3.bn.weight", "mask_estimators.0.segm.hyperace.high_order_branch.2.cv3.bn.bias", "mask_estimators.0.segm.hyperace.high_order_fuse.conv.weight", "mask_estimators.0.segm.hyperace.high_order_fuse.bn.weight", "mask_estimators.0.segm.hyperace.high_order_fuse.bn.bias", "mask_estimators.0.segm.hyperace.low_order_branch.0.cv1.conv.weight", "mask_estimators.0.segm.hyperace.low_order_branch.0.cv1.bn.weight", "mask_estimators.0.segm.hyperace.low_order_branch.0.cv1.bn.bias", "mask_estimators.0.segm.hyperace.low_order_branch.0.cv2.conv.weight", "mask_estimators.0.segm.hyperace.low_order_branch.0.cv2.bn.weight", "mask_estimators.0.segm.hyperace.low_order_branch.0.cv2.bn.bias", "mask_estimators.0.segm.hyperace.low_order_branch.0.cv3.conv.weight", "mask_estimators.0.segm.hyperace.low_order_branch.0.cv3.bn.weight", "mask_estimators.0.segm.hyperace.low_order_branch.0.cv3.bn.bias", "mask_estimators.0.segm.hyperace.low_order_branch.0.m.0.dsconv1.dwconv.weight", "mask_estimators.0.segm.hyperace.low_order_branch.0.m.0.dsconv1.pwconv.weight", "mask_estimators.0.segm.hyperace.low_order_branch.0.m.0.dsconv1.bn.weight", "mask_estimators.0.segm.hyperace.low_order_branch.0.m.0.dsconv1.bn.bias", "mask_estimators.0.segm.hyperace.low_order_branch.0.m.0.dsconv2.dwconv.weight", "mask_estimators.0.segm.hyperace.low_order_branch.0.m.0.dsconv2.pwconv.weight", "mask_estimators.0.segm.hyperace.low_order_branch.0.m.0.dsconv2.bn.weight", "mask_estimators.0.segm.hyperace.low_order_branch.0.m.0.dsconv2.bn.bias", "mask_estimators.0.segm.hyperace.low_order_branch.1.cv1.conv.weight", "mask_estimators.0.segm.hyperace.low_order_branch.1.cv1.bn.weight", "mask_estimators.0.segm.hyperace.low_order_branch.1.cv1.bn.bias", "mask_estimators.0.segm.hyperace.low_order_branch.1.cv2.conv.weight", "mask_estimators.0.segm.hyperace.low_order_branch.1.cv2.bn.weight", "mask_estimators.0.segm.hyperace.low_order_branch.1.cv2.bn.bias", "mask_estimators.0.segm.hyperace.low_order_branch.1.cv3.conv.weight", "mask_estimators.0.segm.hyperace.low_order_branch.1.cv3.bn.weight", "mask_estimators.0.segm.hyperace.low_order_branch.1.cv3.bn.bias", "mask_estimators.0.segm.hyperace.low_order_branch.1.m.0.dsconv1.dwconv.weight", "mask_estimators.0.segm.hyperace.low_order_branch.1.m.0.dsconv1.pwconv.weight", "mask_estimators.0.segm.hyperace.low_order_branch.1.m.0.dsconv1.bn.weight", "mask_estimators.0.segm.hyperace.low_order_branch.1.m.0.dsconv1.bn.bias", "mask_estimators.0.segm.hyperace.low_order_branch.1.m.0.dsconv2.dwconv.weight", "mask_estimators.0.segm.hyperace.low_order_branch.1.m.0.dsconv2.pwconv.weight", "mask_estimators.0.segm.hyperace.low_order_branch.1.m.0.dsconv2.bn.weight", "mask_estimators.0.segm.hyperace.low_order_branch.1.m.0.dsconv2.bn.bias", "mask_estimators.0.segm.hyperace.final_fuse.conv.weight", "mask_estimators.0.segm.hyperace.final_fuse.bn.weight", "mask_estimators.0.segm.hyperace.final_fuse.bn.bias", "mask_estimators.0.segm.decoder.h_to_d5.conv.weight", "mask_estimators.0.segm.decoder.h_to_d5.bn.weight", "mask_estimators.0.segm.decoder.h_to_d5.bn.bias", "mask_estimators.0.segm.decoder.h_to_d4.conv.weight", "mask_estimators.0.segm.decoder.h_to_d4.bn.weight", "mask_estimators.0.segm.decoder.h_to_d4.bn.bias", "mask_estimators.0.segm.decoder.h_to_d3.conv.weight", "mask_estimators.0.segm.decoder.h_to_d3.bn.weight", "mask_estimators.0.segm.decoder.h_to_d3.bn.bias", "mask_estimators.0.segm.decoder.h_to_d2.conv.weight", "mask_estimators.0.segm.decoder.h_to_d2.bn.weight", "mask_estimators.0.segm.decoder.h_to_d2.bn.bias", "mask_estimators.0.segm.decoder.fusion_d5.gamma", "mask_estimators.0.segm.decoder.fusion_d4.gamma", "mask_estimators.0.segm.decoder.fusion_d3.gamma", "mask_estimators.0.segm.decoder.fusion_d2.gamma", "mask_estimators.0.segm.decoder.skip_p5.conv.weight", "mask_estimators.0.segm.decoder.skip_p5.bn.weight", "mask_estimators.0.segm.decoder.skip_p5.bn.bias", "mask_estimators.0.segm.decoder.skip_p4.conv.weight", "mask_estimators.0.segm.decoder.skip_p4.bn.weight", "mask_estimators.0.segm.decoder.skip_p4.bn.bias", "mask_estimators.0.segm.decoder.skip_p3.conv.weight", "mask_estimators.0.segm.decoder.skip_p3.bn.weight", "mask_estimators.0.segm.decoder.skip_p3.bn.bias", "mask_estimators.0.segm.decoder.skip_p2.conv.weight", "mask_estimators.0.segm.decoder.skip_p2.bn.weight", "mask_estimators.0.segm.decoder.skip_p2.bn.bias", "mask_estimators.0.segm.decoder.up_d5.cv1.conv.weight", "mask_estimators.0.segm.decoder.up_d5.cv1.bn.weight", "mask_estimators.0.segm.decoder.up_d5.cv1.bn.bias", "mask_estimators.0.segm.decoder.up_d5.m.cv1.conv.weight", "mask_estimators.0.segm.decoder.up_d5.m.cv1.bn.weight", "mask_estimators.0.segm.decoder.up_d5.m.cv1.bn.bias", "mask_estimators.0.segm.decoder.up_d5.m.cv2.conv.weight", "mask_estimators.0.segm.decoder.up_d5.m.cv2.bn.weight", "mask_estimators.0.segm.decoder.up_d5.m.cv2.bn.bias", "mask_estimators.0.segm.decoder.up_d5.m.cv3.conv.weight", "mask_estimators.0.segm.decoder.up_d5.m.cv3.bn.weight", "mask_estimators.0.segm.decoder.up_d5.m.cv3.bn.bias", "mask_estimators.0.segm.decoder.up_d5.m.m.0.dsconv1.dwconv.weight", "mask_estimators.0.segm.decoder.up_d5.m.m.0.dsconv1.pwconv.weight", "mask_estimators.0.segm.decoder.up_d5.m.m.0.dsconv1.bn.weight", "mask_estimators.0.segm.decoder.up_d5.m.m.0.dsconv1.bn.bias", "mask_estimators.0.segm.decoder.up_d5.m.m.0.dsconv2.dwconv.weight", "mask_estimators.0.segm.decoder.up_d5.m.m.0.dsconv2.pwconv.weight", "mask_estimators.0.segm.decoder.up_d5.m.m.0.dsconv2.bn.weight", "mask_estimators.0.segm.decoder.up_d5.m.m.0.dsconv2.bn.bias", "mask_estimators.0.segm.decoder.up_d5.cv2.conv.weight", "mask_estimators.0.segm.decoder.up_d5.cv2.bn.weight", "mask_estimators.0.segm.decoder.up_d5.cv2.bn.bias", "mask_estimators.0.segm.decoder.up_d4.cv1.conv.weight", "mask_estimators.0.segm.decoder.up_d4.cv1.bn.weight", "mask_estimators.0.segm.decoder.up_d4.cv1.bn.bias", "mask_estimators.0.segm.decoder.up_d4.m.cv1.conv.weight", "mask_estimators.0.segm.decoder.up_d4.m.cv1.bn.weight", "mask_estimators.0.segm.decoder.up_d4.m.cv1.bn.bias", "mask_estimators.0.segm.decoder.up_d4.m.cv2.conv.weight", "mask_estimators.0.segm.decoder.up_d4.m.cv2.bn.weight", "mask_estimators.0.segm.decoder.up_d4.m.cv2.bn.bias", "mask_estimators.0.segm.decoder.up_d4.m.cv3.conv.weight", "mask_estimators.0.segm.decoder.up_d4.m.cv3.bn.weight", "mask_estimators.0.segm.decoder.up_d4.m.cv3.bn.bias", "mask_estimators.0.segm.decoder.up_d4.m.m.0.dsconv1.dwconv.weight", "mask_estimators.0.segm.decoder.up_d4.m.m.0.dsconv1.pwconv.weight", "mask_estimators.0.segm.decoder.up_d4.m.m.0.dsconv1.bn.weight", "mask_estimators.0.segm.decoder.up_d4.m.m.0.dsconv1.bn.bias", "mask_estimators.0.segm.decoder.up_d4.m.m.0.dsconv2.dwconv.weight", "mask_estimators.0.segm.decoder.up_d4.m.m.0.dsconv2.pwconv.weight", "mask_estimators.0.segm.decoder.up_d4.m.m.0.dsconv2.bn.weight", "mask_estimators.0.segm.decoder.up_d4.m.m.0.dsconv2.bn.bias", "mask_estimators.0.segm.decoder.up_d4.cv2.conv.weight", "mask_estimators.0.segm.decoder.up_d4.cv2.bn.weight", "mask_estimators.0.segm.decoder.up_d4.cv2.bn.bias", "mask_estimators.0.segm.decoder.up_d3.cv1.conv.weight", "mask_estimators.0.segm.decoder.up_d3.cv1.bn.weight", "mask_estimators.0.segm.decoder.up_d3.cv1.bn.bias", "mask_estimators.0.segm.decoder.up_d3.m.cv1.conv.weight", "mask_estimators.0.segm.decoder.up_d3.m.cv1.bn.weight", "mask_estimators.0.segm.decoder.up_d3.m.cv1.bn.bias", "mask_estimators.0.segm.decoder.up_d3.m.cv2.conv.weight", "mask_estimators.0.segm.decoder.up_d3.m.cv2.bn.weight", "mask_estimators.0.segm.decoder.up_d3.m.cv2.bn.bias", "mask_estimators.0.segm.decoder.up_d3.m.cv3.conv.weight", "mask_estimators.0.segm.decoder.up_d3.m.cv3.bn.weight", "mask_estimators.0.segm.decoder.up_d3.m.cv3.bn.bias", "mask_estimators.0.segm.decoder.up_d3.m.m.0.dsconv1.dwconv.weight", "mask_estimators.0.segm.decoder.up_d3.m.m.0.dsconv1.pwconv.weight", "mask_estimators.0.segm.decoder.up_d3.m.m.0.dsconv1.bn.weight", "mask_estimators.0.segm.decoder.up_d3.m.m.0.dsconv1.bn.bias", "mask_estimators.0.segm.decoder.up_d3.m.m.0.dsconv2.dwconv.weight", "mask_estimators.0.segm.decoder.up_d3.m.m.0.dsconv2.pwconv.weight", "mask_estimators.0.segm.decoder.up_d3.m.m.0.dsconv2.bn.weight", "mask_estimators.0.segm.decoder.up_d3.m.m.0.dsconv2.bn.bias", "mask_estimators.0.segm.decoder.up_d3.cv2.conv.weight", "mask_estimators.0.segm.decoder.up_d3.cv2.bn.weight", "mask_estimators.0.segm.decoder.up_d3.cv2.bn.bias", "mask_estimators.0.segm.decoder.final_d2.cv1.conv.weight", "mask_estimators.0.segm.decoder.final_d2.cv1.bn.weight", "mask_estimators.0.segm.decoder.final_d2.cv1.bn.bias", "mask_estimators.0.segm.decoder.final_d2.m.cv1.conv.weight", "mask_estimators.0.segm.decoder.final_d2.m.cv1.bn.weight", "mask_estimators.0.segm.decoder.final_d2.m.cv1.bn.bias", "mask_estimators.0.segm.decoder.final_d2.m.cv2.conv.weight", "mask_estimators.0.segm.decoder.final_d2.m.cv2.bn.weight", "mask_estimators.0.segm.decoder.final_d2.m.cv2.bn.bias", "mask_estimators.0.segm.decoder.final_d2.m.cv3.conv.weight", "mask_estimators.0.segm.decoder.final_d2.m.cv3.bn.weight", "mask_estimators.0.segm.decoder.final_d2.m.cv3.bn.bias", "mask_estimators.0.segm.decoder.final_d2.m.m.0.dsconv1.dwconv.weight", "mask_estimators.0.segm.decoder.final_d2.m.m.0.dsconv1.pwconv.weight", "mask_estimators.0.segm.decoder.final_d2.m.m.0.dsconv1.bn.weight", "mask_estimators.0.segm.decoder.final_d2.m.m.0.dsconv1.bn.bias", "mask_estimators.0.segm.decoder.final_d2.m.m.0.dsconv2.dwconv.weight", "mask_estimators.0.segm.decoder.final_d2.m.m.0.dsconv2.pwconv.weight", "mask_estimators.0.segm.decoder.final_d2.m.m.0.dsconv2.bn.weight", "mask_estimators.0.segm.decoder.final_d2.m.m.0.dsconv2.bn.bias", "mask_estimators.0.segm.decoder.final_d2.cv2.conv.weight", "mask_estimators.0.segm.decoder.final_d2.cv2.bn.weight", "mask_estimators.0.segm.decoder.final_d2.cv2.bn.bias", "mask_estimators.0.segm.upsample_head.block1.conv.dwconv.weight", "mask_estimators.0.segm.upsample_head.block1.conv.pwconv.weight", "mask_estimators.0.segm.upsample_head.block1.conv.bn.weight", "mask_estimators.0.segm.upsample_head.block1.conv.bn.bias", "mask_estimators.0.segm.upsample_head.block2.conv.dwconv.weight", "mask_estimators.0.segm.upsample_head.block2.conv.pwconv.weight", "mask_estimators.0.segm.upsample_head.block2.conv.bn.weight", "mask_estimators.0.segm.upsample_head.block2.conv.bn.bias", "mask_estimators.0.segm.upsample_head.block3.conv.dwconv.weight", "mask_estimators.0.segm.upsample_head.block3.conv.pwconv.weight", "mask_estimators.0.segm.upsample_head.block3.conv.bn.weight", "mask_estimators.0.segm.upsample_head.block3.conv.bn.bias", "mask_estimators.0.segm.upsample_head.block4.conv.dwconv.weight", "mask_estimators.0.segm.upsample_head.block4.conv.pwconv.weight", "mask_estimators.0.segm.upsample_head.block4.conv.bn.weight", "mask_estimators.0.segm.upsample_head.block4.conv.bn.bias", "mask_estimators.0.segm.upsample_head.final_conv.weight".
Traceback (most recent call last):
File "E:\AI_Tools\MSST_WebUI\webui\msst.py", line 183, in run_inference
separator = MSSeparator(
File "E:\AI_Tools\MSST_WebUI\inference\msst_infer.py", line 76, in __init__
self.model, self.config = self.load_model()
File "E:\AI_Tools\MSST_WebUI\inference\msst_infer.py", line 123, in load_model
model.load_state_dict(state_dict)
File "E:\AI_Tools\MSST_WebUI\workenv\Lib\site-packages\torch\nn\modules\module.py", line 2593, in load_state_dict
raise RuntimeError(
RuntimeError: Error(s) in loading state_dict for BSRoformer:
Unexpected key(s) in state_dict: "mask_estimators.0.segm.backbone.stem.dwconv.weight", "mask_estimators.0.segm.backbone.stem.pwconv.weight", "mask_estimators.0.segm.backbone.stem.bn.weight", "mask_estimators.0.segm.backbone.stem.bn.bias", "mask_estimators.0.segm.backbone.p2.0.dwconv.weight", "mask_estimators.0.segm.backbone.p2.0.pwconv.weight", "mask_estimators.0.segm.backbone.p2.0.bn.weight", "mask_estimators.0.segm.backbone.p2.0.bn.bias", "mask_estimators.0.segm.backbone.p2.1.cv1.conv.weight", "mask_estimators.0.segm.backbone.p2.1.cv1.bn.weight", "mask_estimators.0.segm.backbone.p2.1.cv1.bn.bias", "mask_estimators.0.segm.backbone.p2.1.m.cv1.conv.weight", "mask_estimators.0.segm.backbone.p2.1.m.cv1.bn.weight", "mask_estimators.0.segm.backbone.p2.1.m.cv1.bn.bias", "mask_estimators.0.segm.backbone.p2.1.m.cv2.conv.weight", "mask_estimators.0.segm.backbone.p2.1.m.cv2.bn.weight", "mask_estimators.0.segm.backbone.p2.1.m.cv2.bn.bias", "mask_estimators.0.segm.backbone.p2.1.m.cv3.conv.weight", "mask_estimators.0.segm.backbone.p2.1.m.cv3.bn.weight", "mask_estimators.0.segm.backbone.p2.1.m.cv3.bn.bias", "mask_estimators.0.segm.backbone.p2.1.m.m.0.dsconv1.dwconv.weight", "mask_estimators.0.segm.backbone.p2.1.m.m.0.dsconv1.pwconv.weight", "mask_estimators.0.segm.backbone.p2.1.m.m.0.dsconv1.bn.weight", "mask_estimators.0.segm.backbone.p2.1.m.m.0.dsconv1.bn.bias", "mask_estimators.0.segm.backbone.p2.1.m.m.0.dsconv2.dwconv.weight", "mask_estimators.0.segm.backbone.p2.1.m.m.0.dsconv2.pwconv.weight", "mask_estimators.0.segm.backbone.p2.1.m.m.0.dsconv2.bn.weight", "mask_estimators.0.segm.backbone.p2.1.m.m.0.dsconv2.bn.bias", "mask_estimators.0.segm.backbone.p2.1.m.m.1.dsconv1.dwconv.weight", "mask_estimators.0.segm.backbone.p2.1.m.m.1.dsconv1.pwconv.weight", "mask_estimators.0.segm.backbone.p2.1.m.m.1.dsconv1.bn.weight", "mask_estimators.0.segm.backbone.p2.1.m.m.1.dsconv1.bn.bias", "mask_estimators.0.segm.backbone.p2.1.m.m.1.dsconv2.dwconv.weight", "mask_estimators.0.segm.backbone.p2.1.m.m.1.dsconv2.pwconv.weight", "mask_estimators.0.segm.backbone.p2.1.m.m.1.dsconv2.bn.weight", "mask_estimators.0.segm.backbone.p2.1.m.m.1.dsconv2.bn.bias", "mask_estimators.0.segm.backbone.p2.1.cv2.conv.weight", "mask_estimators.0.segm.backbone.p2.1.cv2.bn.weight", "mask_estimators.0.segm.backbone.p2.1.cv2.bn.bias", "mask_estimators.0.segm.backbone.p3.0.dwconv.weight", "mask_estimators.0.segm.backbone.p3.0.pwconv.weight", "mask_estimators.0.segm.backbone.p3.0.bn.weight", "mask_estimators.0.segm.backbone.p3.0.bn.bias", "mask_estimators.0.segm.backbone.p3.1.cv1.conv.weight", "mask_estimators.0.segm.backbone.p3.1.cv1.bn.weight", "mask_estimators.0.segm.backbone.p3.1.cv1.bn.bias", "mask_estimators.0.segm.backbone.p3.1.m.cv1.conv.weight", "mask_estimators.0.segm.backbone.p3.1.m.cv1.bn.weight", "mask_estimators.0.segm.backbone.p3.1.m.cv1.bn.bias", "mask_estimators.0.segm.backbone.p3.1.m.cv2.conv.weight", "mask_estimators.0.segm.backbone.p3.1.m.cv2.bn.weight", "mask_estimators.0.segm.backbone.p3.1.m.cv2.bn.bias", "mask_estimators.0.segm.backbone.p3.1.m.cv3.conv.weight", "mask_estimators.0.segm.backbone.p3.1.m.cv3.bn.weight", "mask_estimators.0.segm.backbone.p3.1.m.cv3.bn.bias", "mask_estimators.0.segm.backbone.p3.1.m.m.0.dsconv1.dwconv.weight", "mask_estimators.0.segm.backbone.p3.1.m.m.0.dsconv1.pwconv.weight", "mask_estimators.0.segm.backbone.p3.1.m.m.0.dsconv1.bn.weight", "mask_estimators.0.segm.backbone.p3.1.m.m.0.dsconv1.bn.bias", "mask_estimators.0.segm.backbone.p3.1.m.m.0.dsconv2.dwconv.weight", "mask_estimators.0.segm.backbone.p3.1.m.m.0.dsconv2.pwconv.weight", "mask_estimators.0.segm.backbone.p3.1.m.m.0.dsconv2.bn.weight", "mask_estimators.0.segm.backbone.p3.1.m.m.0.dsconv2.bn.bias", "mask_estimators.0.segm.backbone.p3.1.m.m.1.dsconv1.dwconv.weight", "mask_estimators.0.segm.backbone.p3.1.m.m.1.dsconv1.pwconv.weight", "mask_estimators.0.segm.backbone.p3.1.m.m.1.dsconv1.bn.weight", "mask_estimators.0.segm.backbone.p3.1.m.m.1.dsconv1.bn.bias", "mask_estimators.0.segm.backbone.p3.1.m.m.1.dsconv2.dwconv.weight", "mask_estimators.0.segm.backbone.p3.1.m.m.1.dsconv2.pwconv.weight", "mask_estimators.0.segm.backbone.p3.1.m.m.1.dsconv2.bn.weight", "mask_estimators.0.segm.backbone.p3.1.m.m.1.dsconv2.bn.bias", "mask_estimators.0.segm.backbone.p3.1.m.m.2.dsconv1.dwconv.weight", "mask_estimators.0.segm.backbone.p3.1.m.m.2.dsconv1.pwconv.weight", "mask_estimators.0.segm.backbone.p3.1.m.m.2.dsconv1.bn.weight", "mask_estimators.0.segm.backbone.p3.1.m.m.2.dsconv1.bn.bias", "mask_estimators.0.segm.backbone.p3.1.m.m.2.dsconv2.dwconv.weight", "mask_estimators.0.segm.backbone.p3.1.m.m.2.dsconv2.pwconv.weight", "mask_estimators.0.segm.backbone.p3.1.m.m.2.dsconv2.bn.weight", "mask_estimators.0.segm.backbone.p3.1.m.m.2.dsconv2.bn.bias", "mask_estimators.0.segm.backbone.p3.1.m.m.3.dsconv1.dwconv.weight", "mask_estimators.0.segm.backbone.p3.1.m.m.3.dsconv1.pwconv.weight", "mask_estimators.0.segm.backbone.p3.1.m.m.3.dsconv1.bn.weight", "mask_estimators.0.segm.backbone.p3.1.m.m.3.dsconv1.bn.bias", "mask_estimators.0.segm.backbone.p3.1.m.m.3.dsconv2.dwconv.weight", "mask_estimators.0.segm.backbone.p3.1.m.m.3.dsconv2.pwconv.weight", "mask_estimators.0.segm.backbone.p3.1.m.m.3.dsconv2.bn.weight", "mask_estimators.0.segm.backbone.p3.1.m.m.3.dsconv2.bn.bias", "mask_estimators.0.segm.backbone.p3.1.cv2.conv.weight", "mask_estimators.0.segm.backbone.p3.1.cv2.bn.weight", "mask_estimators.0.segm.backbone.p3.1.cv2.bn.bias", "mask_estimators.0.segm.backbone.p4.0.dwconv.weight", "mask_estimators.0.segm.backbone.p4.0.pwconv.weight", "mask_estimators.0.segm.backbone.p4.0.bn.weight", "mask_estimators.0.segm.backbone.p4.0.bn.bias", "mask_estimators.0.segm.backbone.p4.1.cv1.conv.weight", "mask_estimators.0.segm.backbone.p4.1.cv1.bn.weight", "mask_estimators.0.segm.backbone.p4.1.cv1.bn.bias", "mask_estimators.0.segm.backbone.p4.1.m.cv1.conv.weight", "mask_estimators.0.segm.backbone.p4.1.m.cv1.bn.weight", "mask_estimators.0.segm.backbone.p4.1.m.cv1.bn.bias", "mask_estimators.0.segm.backbone.p4.1.m.cv2.conv.weight", "mask_estimators.0.segm.backbone.p4.1.m.cv2.bn.weight", "mask_estimators.0.segm.backbone.p4.1.m.cv2.bn.bias", "mask_estimators.0.segm.backbone.p4.1.m.cv3.conv.weight", "mask_estimators.0.segm.backbone.p4.1.m.cv3.bn.weight", "mask_estimators.0.segm.backbone.p4.1.m.cv3.bn.bias", "mask_estimators.0.segm.backbone.p4.1.m.m.0.dsconv1.dwconv.weight", "mask_estimators.0.segm.backbone.p4.1.m.m.0.dsconv1.pwconv.weight", "mask_estimators.0.segm.backbone.p4.1.m.m.0.dsconv1.bn.weight", "mask_estimators.0.segm.backbone.p4.1.m.m.0.dsconv1.bn.bias", "mask_estimators.0.segm.backbone.p4.1.m.m.0.dsconv2.dwconv.weight", "mask_estimators.0.segm.backbone.p4.1.m.m.0.dsconv2.pwconv.weight", "mask_estimators.0.segm.backbone.p4.1.m.m.0.dsconv2.bn.weight", "mask_estimators.0.segm.backbone.p4.1.m.m.0.dsconv2.bn.bias", "mask_estimators.0.segm.backbone.p4.1.m.m.1.dsconv1.dwconv.weight", "mask_estimators.0.segm.backbone.p4.1.m.m.1.dsconv1.pwconv.weight", "mask_estimators.0.segm.backbone.p4.1.m.m.1.dsconv1.bn.weight", "mask_estimators.0.segm.backbone.p4.1.m.m.1.dsconv1.bn.bias", "mask_estimators.0.segm.backbone.p4.1.m.m.1.dsconv2.dwconv.weight", "mask_estimators.0.segm.backbone.p4.1.m.m.1.dsconv2.pwconv.weight", "mask_estimators.0.segm.backbone.p4.1.m.m.1.dsconv2.bn.weight", "mask_estimators.0.segm.backbone.p4.1.m.m.1.dsconv2.bn.bias", "mask_estimators.0.segm.backbone.p4.1.m.m.2.dsconv1.dwconv.weight", "mask_estimators.0.segm.backbone.p4.1.m.m.2.dsconv1.pwconv.weight", "mask_estimators.0.segm.backbone.p4.1.m.m.2.dsconv1.bn.weight", "mask_estimators.0.segm.backbone.p4.1.m.m.2.dsconv1.bn.bias", "mask_estimators.0.segm.backbone.p4.1.m.m.2.dsconv2.dwconv.weight", "mask_estimators.0.segm.backbone.p4.1.m.m.2.dsconv2.pwconv.weight", "mask_estimators.0.segm.backbone.p4.1.m.m.2.dsconv2.bn.weight", "mask_estimators.0.segm.backbone.p4.1.m.m.2.dsconv2.bn.bias", "mask_estimators.0.segm.backbone.p4.1.m.m.3.dsconv1.dwconv.weight", "mask_estimators.0.segm.backbone.p4.1.m.m.3.dsconv1.pwconv.weight", "mask_estimators.0.segm.backbone.p4.1.m.m.3.dsconv1.bn.weight", "mask_estimators.0.segm.backbone.p4.1.m.m.3.dsconv1.bn.bias", "mask_estimators.0.segm.backbone.p4.1.m.m.3.dsconv2.dwconv.weight", "mask_estimators.0.segm.backbone.p4.1.m.m.3.dsconv2.pwconv.weight", "mask_estimators.0.segm.backbone.p4.1.m.m.3.dsconv2.bn.weight", "mask_estimators.0.segm.backbone.p4.1.m.m.3.dsconv2.bn.bias", "mask_estimators.0.segm.backbone.p4.1.cv2.conv.weight", "mask_estimators.0.segm.backbone.p4.1.cv2.bn.weight", "mask_estimators.0.segm.backbone.p4.1.cv2.bn.bias", "mask_estimators.0.segm.backbone.p5.0.dwconv.weight", "mask_estimators.0.segm.backbone.p5.0.pwconv.weight", "mask_estimators.0.segm.backbone.p5.0.bn.weight", "mask_estimators.0.segm.backbone.p5.0.bn.bias", "mask_estimators.0.segm.backbone.p5.1.cv1.conv.weight", "mask_estimators.0.segm.backbone.p5.1.cv1.bn.weight", "mask_estimators.0.segm.backbone.p5.1.cv1.bn.bias", "mask_estimators.0.segm.backbone.p5.1.m.cv1.conv.weight", "mask_estimators.0.segm.backbone.p5.1.m.cv1.bn.weight", "mask_estimators.0.segm.backbone.p5.1.m.cv1.bn.bias", "mask_estimators.0.segm.backbone.p5.1.m.cv2.conv.weight", "mask_estimators.0.segm.backbone.p5.1.m.cv2.bn.weight", "mask_estimators.0.segm.backbone.p5.1.m.cv2.bn.bias", "mask_estimators.0.segm.backbone.p5.1.m.cv3.conv.weight", "mask_estimators.0.segm.backbone.p5.1.m.cv3.bn.weight", "mask_estimators.0.segm.backbone.p5.1.m.cv3.bn.bias", "mask_estimators.0.segm.backbone.p5.1.m.m.0.dsconv1.dwconv.weight", "mask_estimators.0.segm.backbone.p5.1.m.m.0.dsconv1.pwconv.weight", "mask_estimators.0.segm.backbone.p5.1.m.m.0.dsconv1.bn.weight", "mask_estimators.0.segm.backbone.p5.1.m.m.0.dsconv1.bn.bias", "mask_estimators.0.segm.backbone.p5.1.m.m.0.dsconv2.dwconv.weight", "mask_estimators.0.segm.backbone.p5.1.m.m.0.dsconv2.pwconv.weight", "mask_estimators.0.segm.backbone.p5.1.m.m.0.dsconv2.bn.weight", "mask_estimators.0.segm.backbone.p5.1.m.m.0.dsconv2.bn.bias", "mask_estimators.0.segm.backbone.p5.1.m.m.1.dsconv1.dwconv.weight", "mask_estimators.0.segm.backbone.p5.1.m.m.1.dsconv1.pwconv.weight", "mask_estimators.0.segm.backbone.p5.1.m.m.1.dsconv1.bn.weight", "mask_estimators.0.segm.backbone.p5.1.m.m.1.dsconv1.bn.bias", "mask_estimators.0.segm.backbone.p5.1.m.m.1.dsconv2.dwconv.weight", "mask_estimators.0.segm.backbone.p5.1.m.m.1.dsconv2.pwconv.weight", "mask_estimators.0.segm.backbone.p5.1.m.m.1.dsconv2.bn.weight", "mask_estimators.0.segm.backbone.p5.1.m.m.1.dsconv2.bn.bias", "mask_estimators.0.segm.backbone.p5.1.cv2.conv.weight", "mask_estimators.0.segm.backbone.p5.1.cv2.bn.weight", "mask_estimators.0.segm.backbone.p5.1.cv2.bn.bias", "mask_estimators.0.segm.hyperace.fuse_conv.conv.weight", "mask_estimators.0.segm.hyperace.fuse_conv.bn.weight", "mask_estimators.0.segm.hyperace.fuse_conv.bn.bias", "mask_estimators.0.segm.hyperace.high_order_branch.0.cv1.conv.weight", "mask_estimators.0.segm.hyperace.high_order_branch.0.cv1.bn.weight", "mask_estimators.0.segm.hyperace.high_order_branch.0.cv1.bn.bias", "mask_estimators.0.segm.hyperace.high_order_branch.0.cv2.conv.weight", "mask_estimators.0.segm.hyperace.high_order_branch.0.cv2.bn.weight", "mask_estimators.0.segm.hyperace.high_order_branch.0.cv2.bn.bias", "mask_estimators.0.segm.hyperace.high_order_branch.0.ahc.adaptive_hyperedge_gen.global_proto", "mask_estimators.0.segm.hyperace.high_order_branch.0.ahc.adaptive_hyperedge_gen.context_mapper.weight", "mask_estimators.0.segm.hyperace.high_order_branch.0.ahc.adaptive_hyperedge_gen.query_proj.weight", "mask_estimators.0.segm.hyperace.high_order_branch.0.ahc.hypergraph_conv.W_e.weight", "mask_estimators.0.segm.hyperace.high_order_branch.0.ahc.hypergraph_conv.W_v.weight", "mask_estimators.0.segm.hyperace.high_order_branch.0.cv3.conv.weight", "mask_estimators.0.segm.hyperace.high_order_branch.0.cv3.bn.weight", "mask_estimators.0.segm.hyperace.high_order_branch.0.cv3.bn.bias", "mask_estimators.0.segm.hyperace.high_order_branch.1.cv1.conv.weight", "mask_estimators.0.segm.hyperace.high_order_branch.1.cv1.bn.weight", "mask_estimators.0.segm.hyperace.high_order_branch.1.cv1.bn.bias", "mask_estimators.0.segm.hyperace.high_order_branch.1.cv2.conv.weight", "mask_estimators.0.segm.hyperace.high_order_branch.1.cv2.bn.weight", "mask_estimators.0.segm.hyperace.high_order_branch.1.cv2.bn.bias", "mask_estimators.0.segm.hyperace.high_order_branch.1.ahc.adaptive_hyperedge_gen.global_proto", "mask_estimators.0.segm.hyperace.high_order_branch.1.ahc.adaptive_hyperedge_gen.context_mapper.weight", "mask_estimators.0.segm.hyperace.high_order_branch.1.ahc.adaptive_hyperedge_gen.query_proj.weight", "mask_estimators.0.segm.hyperace.high_order_branch.1.ahc.hypergraph_conv.W_e.weight", "mask_estimators.0.segm.hyperace.high_order_branch.1.ahc.hypergraph_conv.W_v.weight", "mask_estimators.0.segm.hyperace.high_order_branch.1.cv3.conv.weight", "mask_estimators.0.segm.hyperace.high_order_branch.1.cv3.bn.weight", "mask_estimators.0.segm.hyperace.high_order_branch.1.cv3.bn.bias", "mask_estimators.0.segm.hyperace.high_order_branch.2.cv1.conv.weight", "mask_estimators.0.segm.hyperace.high_order_branch.2.cv1.bn.weight", "mask_estimators.0.segm.hyperace.high_order_branch.2.cv1.bn.bias", "mask_estimators.0.segm.hyperace.high_order_branch.2.cv2.conv.weight", "mask_estimators.0.segm.hyperace.high_order_branch.2.cv2.bn.weight", "mask_estimators.0.segm.hyperace.high_order_branch.2.cv2.bn.bias", "mask_estimators.0.segm.hyperace.high_order_branch.2.ahc.adaptive_hyperedge_gen.global_proto", "mask_estimators.0.segm.hyperace.high_order_branch.2.ahc.adaptive_hyperedge_gen.context_mapper.weight", "mask_estimators.0.segm.hyperace.high_order_branch.2.ahc.adaptive_hyperedge_gen.query_proj.weight", "mask_estimators.0.segm.hyperace.high_order_branch.2.ahc.hypergraph_conv.W_e.weight", "mask_estimators.0.segm.hyperace.high_order_branch.2.ahc.hypergraph_conv.W_v.weight", "mask_estimators.0.segm.hyperace.high_order_branch.2.cv3.conv.weight", "mask_estimators.0.segm.hyperace.high_order_branch.2.cv3.bn.weight", "mask_estimators.0.segm.hyperace.high_order_branch.2.cv3.bn.bias", "mask_estimators.0.segm.hyperace.high_order_fuse.conv.weight", "mask_estimators.0.segm.hyperace.high_order_fuse.bn.weight", "mask_estimators.0.segm.hyperace.high_order_fuse.bn.bias", "mask_estimators.0.segm.hyperace.low_order_branch.0.cv1.conv.weight", "mask_estimators.0.segm.hyperace.low_order_branch.0.cv1.bn.weight", "mask_estimators.0.segm.hyperace.low_order_branch.0.cv1.bn.bias", "mask_estimators.0.segm.hyperace.low_order_branch.0.cv2.conv.weight", "mask_estimators.0.segm.hyperace.low_order_branch.0.cv2.bn.weight", "mask_estimators.0.segm.hyperace.low_order_branch.0.cv2.bn.bias", "mask_estimators.0.segm.hyperace.low_order_branch.0.cv3.conv.weight", "mask_estimators.0.segm.hyperace.low_order_branch.0.cv3.bn.weight", "mask_estimators.0.segm.hyperace.low_order_branch.0.cv3.bn.bias", "mask_estimators.0.segm.hyperace.low_order_branch.0.m.0.dsconv1.dwconv.weight", "mask_estimators.0.segm.hyperace.low_order_branch.0.m.0.dsconv1.pwconv.weight", "mask_estimators.0.segm.hyperace.low_order_branch.0.m.0.dsconv1.bn.weight", "mask_estimators.0.segm.hyperace.low_order_branch.0.m.0.dsconv1.bn.bias", "mask_estimators.0.segm.hyperace.low_order_branch.0.m.0.dsconv2.dwconv.weight", "mask_estimators.0.segm.hyperace.low_order_branch.0.m.0.dsconv2.pwconv.weight", "mask_estimators.0.segm.hyperace.low_order_branch.0.m.0.dsconv2.bn.weight", "mask_estimators.0.segm.hyperace.low_order_branch.0.m.0.dsconv2.bn.bias", "mask_estimators.0.segm.hyperace.low_order_branch.1.cv1.conv.weight", "mask_estimators.0.segm.hyperace.low_order_branch.1.cv1.bn.weight", "mask_estimators.0.segm.hyperace.low_order_branch.1.cv1.bn.bias", "mask_estimators.0.segm.hyperace.low_order_branch.1.cv2.conv.weight", "mask_estimators.0.segm.hyperace.low_order_branch.1.cv2.bn.weight", "mask_estimators.0.segm.hyperace.low_order_branch.1.cv2.bn.bias", "mask_estimators.0.segm.hyperace.low_order_branch.1.cv3.conv.weight", "mask_estimators.0.segm.hyperace.low_order_branch.1.cv3.bn.weight", "mask_estimators.0.segm.hyperace.low_order_branch.1.cv3.bn.bias", "mask_estimators.0.segm.hyperace.low_order_branch.1.m.0.dsconv1.dwconv.weight", "mask_estimators.0.segm.hyperace.low_order_branch.1.m.0.dsconv1.pwconv.weight", "mask_estimators.0.segm.hyperace.low_order_branch.1.m.0.dsconv1.bn.weight", "mask_estimators.0.segm.hyperace.low_order_branch.1.m.0.dsconv1.bn.bias", "mask_estimators.0.segm.hyperace.low_order_branch.1.m.0.dsconv2.dwconv.weight", "mask_estimators.0.segm.hyperace.low_order_branch.1.m.0.dsconv2.pwconv.weight", "mask_estimators.0.segm.hyperace.low_order_branch.1.m.0.dsconv2.bn.weight", "mask_estimators.0.segm.hyperace.low_order_branch.1.m.0.dsconv2.bn.bias", "mask_estimators.0.segm.hyperace.final_fuse.conv.weight", "mask_estimators.0.segm.hyperace.final_fuse.bn.weight", "mask_estimators.0.segm.hyperace.final_fuse.bn.bias", "mask_estimators.0.segm.decoder.h_to_d5.conv.weight", "mask_estimators.0.segm.decoder.h_to_d5.bn.weight", "mask_estimators.0.segm.decoder.h_to_d5.bn.bias", "mask_estimators.0.segm.decoder.h_to_d4.conv.weight", "mask_estimators.0.segm.decoder.h_to_d4.bn.weight", "mask_estimators.0.segm.decoder.h_to_d4.bn.bias", "mask_estimators.0.segm.decoder.h_to_d3.conv.weight", "mask_estimators.0.segm.decoder.h_to_d3.bn.weight", "mask_estimators.0.segm.decoder.h_to_d3.bn.bias", "mask_estimators.0.segm.decoder.h_to_d2.conv.weight", "mask_estimators.0.segm.decoder.h_to_d2.bn.weight", "mask_estimators.0.segm.decoder.h_to_d2.bn.bias", "mask_estimators.0.segm.decoder.fusion_d5.gamma", "mask_estimators.0.segm.decoder.fusion_d4.gamma", "mask_estimators.0.segm.decoder.fusion_d3.gamma", "mask_estimators.0.segm.decoder.fusion_d2.gamma", "mask_estimators.0.segm.decoder.skip_p5.conv.weight", "mask_estimators.0.segm.decoder.skip_p5.bn.weight", "mask_estimators.0.segm.decoder.skip_p5.bn.bias", "mask_estimators.0.segm.decoder.skip_p4.conv.weight", "mask_estimators.0.segm.decoder.skip_p4.bn.weight", "mask_estimators.0.segm.decoder.skip_p4.bn.bias", "mask_estimators.0.segm.decoder.skip_p3.conv.weight", "mask_estimators.0.segm.decoder.skip_p3.bn.weight", "mask_estimators.0.segm.decoder.skip_p3.bn.bias", "mask_estimators.0.segm.decoder.skip_p2.conv.weight", "mask_estimators.0.segm.decoder.skip_p2.bn.weight", "mask_estimators.0.segm.decoder.skip_p2.bn.bias", "mask_estimators.0.segm.decoder.up_d5.cv1.conv.weight", "mask_estimators.0.segm.decoder.up_d5.cv1.bn.weight", "mask_estimators.0.segm.decoder.up_d5.cv1.bn.bias", "mask_estimators.0.segm.decoder.up_d5.m.cv1.conv.weight", "mask_estimators.0.segm.decoder.up_d5.m.cv1.bn.weight", "mask_estimators.0.segm.decoder.up_d5.m.cv1.bn.bias", "mask_estimators.0.segm.decoder.up_d5.m.cv2.conv.weight", "mask_estimators.0.segm.decoder.up_d5.m.cv2.bn.weight", "mask_estimators.0.segm.decoder.up_d5.m.cv2.bn.bias", "mask_estimators.0.segm.decoder.up_d5.m.cv3.conv.weight", "mask_estimators.0.segm.decoder.up_d5.m.cv3.bn.weight", "mask_estimators.0.segm.decoder.up_d5.m.cv3.bn.bias", "mask_estimators.0.segm.decoder.up_d5.m.m.0.dsconv1.dwconv.weight", "mask_estimators.0.segm.decoder.up_d5.m.m.0.dsconv1.pwconv.weight", "mask_estimators.0.segm.decoder.up_d5.m.m.0.dsconv1.bn.weight", "mask_estimators.0.segm.decoder.up_d5.m.m.0.dsconv1.bn.bias", "mask_estimators.0.segm.decoder.up_d5.m.m.0.dsconv2.dwconv.weight", "mask_estimators.0.segm.decoder.up_d5.m.m.0.dsconv2.pwconv.weight", "mask_estimators.0.segm.decoder.up_d5.m.m.0.dsconv2.bn.weight", "mask_estimators.0.segm.decoder.up_d5.m.m.0.dsconv2.bn.bias", "mask_estimators.0.segm.decoder.up_d5.cv2.conv.weight", "mask_estimators.0.segm.decoder.up_d5.cv2.bn.weight", "mask_estimators.0.segm.decoder.up_d5.cv2.bn.bias", "mask_estimators.0.segm.decoder.up_d4.cv1.conv.weight", "mask_estimators.0.segm.decoder.up_d4.cv1.bn.weight", "mask_estimators.0.segm.decoder.up_d4.cv1.bn.bias", "mask_estimators.0.segm.decoder.up_d4.m.cv1.conv.weight", "mask_estimators.0.segm.decoder.up_d4.m.cv1.bn.weight", "mask_estimators.0.segm.decoder.up_d4.m.cv1.bn.bias", "mask_estimators.0.segm.decoder.up_d4.m.cv2.conv.weight", "mask_estimators.0.segm.decoder.up_d4.m.cv2.bn.weight", "mask_estimators.0.segm.decoder.up_d4.m.cv2.bn.bias", "mask_estimators.0.segm.decoder.up_d4.m.cv3.conv.weight", "mask_estimators.0.segm.decoder.up_d4.m.cv3.bn.weight", "mask_estimators.0.segm.decoder.up_d4.m.cv3.bn.bias", "mask_estimators.0.segm.decoder.up_d4.m.m.0.dsconv1.dwconv.weight", "mask_estimators.0.segm.decoder.up_d4.m.m.0.dsconv1.pwconv.weight", "mask_estimators.0.segm.decoder.up_d4.m.m.0.dsconv1.bn.weight", "mask_estimators.0.segm.decoder.up_d4.m.m.0.dsconv1.bn.bias", "mask_estimators.0.segm.decoder.up_d4.m.m.0.dsconv2.dwconv.weight", "mask_estimators.0.segm.decoder.up_d4.m.m.0.dsconv2.pwconv.weight", "mask_estimators.0.segm.decoder.up_d4.m.m.0.dsconv2.bn.weight", "mask_estimators.0.segm.decoder.up_d4.m.m.0.dsconv2.bn.bias", "mask_estimators.0.segm.decoder.up_d4.cv2.conv.weight", "mask_estimators.0.segm.decoder.up_d4.cv2.bn.weight", "mask_estimators.0.segm.decoder.up_d4.cv2.bn.bias", "mask_estimators.0.segm.decoder.up_d3.cv1.conv.weight", "mask_estimators.0.segm.decoder.up_d3.cv1.bn.weight", "mask_estimators.0.segm.decoder.up_d3.cv1.bn.bias", "mask_estimators.0.segm.decoder.up_d3.m.cv1.conv.weight", "mask_estimators.0.segm.decoder.up_d3.m.cv1.bn.weight", "mask_estimators.0.segm.decoder.up_d3.m.cv1.bn.bias", "mask_estimators.0.segm.decoder.up_d3.m.cv2.conv.weight", "mask_estimators.0.segm.decoder.up_d3.m.cv2.bn.weight", "mask_estimators.0.segm.decoder.up_d3.m.cv2.bn.bias", "mask_estimators.0.segm.decoder.up_d3.m.cv3.conv.weight", "mask_estimators.0.segm.decoder.up_d3.m.cv3.bn.weight", "mask_estimators.0.segm.decoder.up_d3.m.cv3.bn.bias", "mask_estimators.0.segm.decoder.up_d3.m.m.0.dsconv1.dwconv.weight", "mask_estimators.0.segm.decoder.up_d3.m.m.0.dsconv1.pwconv.weight", "mask_estimators.0.segm.decoder.up_d3.m.m.0.dsconv1.bn.weight", "mask_estimators.0.segm.decoder.up_d3.m.m.0.dsconv1.bn.bias", "mask_estimators.0.segm.decoder.up_d3.m.m.0.dsconv2.dwconv.weight", "mask_estimators.0.segm.decoder.up_d3.m.m.0.dsconv2.pwconv.weight", "mask_estimators.0.segm.decoder.up_d3.m.m.0.dsconv2.bn.weight", "mask_estimators.0.segm.decoder.up_d3.m.m.0.dsconv2.bn.bias", "mask_estimators.0.segm.decoder.up_d3.cv2.conv.weight", "mask_estimators.0.segm.decoder.up_d3.cv2.bn.weight", "mask_estimators.0.segm.decoder.up_d3.cv2.bn.bias", "mask_estimators.0.segm.decoder.final_d2.cv1.conv.weight", "mask_estimators.0.segm.decoder.final_d2.cv1.bn.weight", "mask_estimators.0.segm.decoder.final_d2.cv1.bn.bias", "mask_estimators.0.segm.decoder.final_d2.m.cv1.conv.weight", "mask_estimators.0.segm.decoder.final_d2.m.cv1.bn.weight", "mask_estimators.0.segm.decoder.final_d2.m.cv1.bn.bias", "mask_estimators.0.segm.decoder.final_d2.m.cv2.conv.weight", "mask_estimators.0.segm.decoder.final_d2.m.cv2.bn.weight", "mask_estimators.0.segm.decoder.final_d2.m.cv2.bn.bias", "mask_estimators.0.segm.decoder.final_d2.m.cv3.conv.weight", "mask_estimators.0.segm.decoder.final_d2.m.cv3.bn.weight", "mask_estimators.0.segm.decoder.final_d2.m.cv3.bn.bias", "mask_estimators.0.segm.decoder.final_d2.m.m.0.dsconv1.dwconv.weight", "mask_estimators.0.segm.decoder.final_d2.m.m.0.dsconv1.pwconv.weight", "mask_estimators.0.segm.decoder.final_d2.m.m.0.dsconv1.bn.weight", "mask_estimators.0.segm.decoder.final_d2.m.m.0.dsconv1.bn.bias", "mask_estimators.0.segm.decoder.final_d2.m.m.0.dsconv2.dwconv.weight", "mask_estimators.0.segm.decoder.final_d2.m.m.0.dsconv2.pwconv.weight", "mask_estimators.0.segm.decoder.final_d2.m.m.0.dsconv2.bn.weight", "mask_estimators.0.segm.decoder.final_d2.m.m.0.dsconv2.bn.bias", "mask_estimators.0.segm.decoder.final_d2.cv2.conv.weight", "mask_estimators.0.segm.decoder.final_d2.cv2.bn.weight", "mask_estimators.0.segm.decoder.final_d2.cv2.bn.bias", "mask_estimators.0.segm.upsample_head.block1.conv.dwconv.weight", "mask_estimators.0.segm.upsample_head.block1.conv.pwconv.weight", "mask_estimators.0.segm.upsample_head.block1.conv.bn.weight", "mask_estimators.0.segm.upsample_head.block1.conv.bn.bias", "mask_estimators.0.segm.upsample_head.block2.conv.dwconv.weight", "mask_estimators.0.segm.upsample_head.block2.conv.pwconv.weight", "mask_estimators.0.segm.upsample_head.block2.conv.bn.weight", "mask_estimators.0.segm.upsample_head.block2.conv.bn.bias", "mask_estimators.0.segm.upsample_head.block3.conv.dwconv.weight", "mask_estimators.0.segm.upsample_head.block3.conv.pwconv.weight", "mask_estimators.0.segm.upsample_head.block3.conv.bn.weight", "mask_estimators.0.segm.upsample_head.block3.conv.bn.bias", "mask_estimators.0.segm.upsample_head.block4.conv.dwconv.weight", "mask_estimators.0.segm.upsample_head.block4.conv.pwconv.weight", "mask_estimators.0.segm.upsample_head.block4.conv.bn.weight", "mask_estimators.0.segm.upsample_head.block4.conv.bn.bias", "mask_estimators.0.segm.upsample_head.final_conv.weight".
另外还有一个模型，链接为：pcunwa/BS-Roformer-Inst-FNO
根据模型自述，其修改了掩码估计器(mask estimator)为 FNO1d，需要修改推理代码并安装新依赖，目前版本下报错如下：
21:07:03.683 [ERROR] [webui\msst.py:206] Separation failed: Weights only load failed. This file can still be loaded, to do so you have two options, do those steps only if you trust the source of the checkpoint.
(1) In PyTorch 2.6, we changed the default value of the `weights_only` argument in `torch.load` from `False` to `True`. Re-running `torch.load` with `weights_only` set to `False` will likely succeed, but it can result in arbitrary code execution. Do it only if you got the file from a trusted source.
(2) Alternatively, to load with `weights_only=True` please check the recommended steps in the following error message.
WeightsUnpickler error: Unsupported global: GLOBAL torch._C._nn.gelu was not an allowed global by default. Please use `torch.serialization.add_safe_globals([torch._C._nn.gelu])` or the `torch.serialization.safe_globals([torch._C._nn.gelu])` context manager to allowlist this global if you trust this class/function.
Check the documentation of torch.load to learn more about types accepted by default with weights_only https://pytorch.org/docs/stable/generated/torch.load.html.
Traceback (most recent call last):
File "E:\AI_Tools\MSST_WebUI\webui\msst.py", line 183, in run_inference
separator = MSSeparator(
File "E:\AI_Tools\MSST_WebUI\inference\msst_infer.py", line 76, in __init__
self.model, self.config = self.load_model()
File "E:\AI_Tools\MSST_WebUI\inference\msst_infer.py", line 122, in load_model
state_dict = torch.load(self.model_path, map_location=self.device, weights_only=True)
File "E:\AI_Tools\MSST_WebUI\workenv\Lib\site-packages\torch\serialization.py", line 1524, in load
raise pickle.UnpicklingError(_get_wo_message(str(e))) from None
_pickle.UnpicklingError: Weights only load failed. This file can still be loaded, to do so you have two options, do those steps only if you trust the source of the checkpoint.
(1) In PyTorch 2.6, we changed the default value of the `weights_only` argument in `torch.load` from `False` to `True`. Re-running `torch.load` with `weights_only` set to `False` will likely succeed, but it can result in arbitrary code execution. Do it only if you got the file from a trusted source.
(2) Alternatively, to load with `weights_only=True` please check the recommended steps in the following error message.
WeightsUnpickler error: Unsupported global: GLOBAL torch._C._nn.gelu was not an allowed global by default. Please use `torch.serialization.add_safe_globals([torch._C._nn.gelu])` or the `torch.serialization.safe_globals([torch._C._nn.gelu])` context manager to allowlist this global if you trust this class/function.
Check the documentation of torch.load to learn more about types accepted by default with weights_only https://pytorch.org/docs/stable/generated/torch.load.html.
若无适配计划，请告知并关闭issue，感谢。