feat: Support SDPose-OOD (#12661)

This commit is contained in:
Jukka Seppänen
2026-02-27 02:59:05 +02:00
committed by GitHub
parent 08b26ed7c2
commit c7f7d52b68
7 changed files with 888 additions and 3 deletions

View File

@@ -525,7 +525,8 @@ class LotusD(SD20):
}
unet_extra_config = {
"num_classes": 'sequential'
"num_classes": 'sequential',
"num_head_channels": 64,
}
def get_model(self, state_dict, prefix="", device=None):