support wan2.2-animate-14b

2026-03-20 23:58:12 +00:00 · 2025-09-30 12:45:56 +08:00
parent 0d6de58af9
commit a36f2f6032
15 changed files with 999 additions and 33 deletions
--- a/diffsynth/trainers/unified_dataset.py
+++ b/diffsynth/trainers/unified_dataset.py
@@ -316,7 +316,7 @@ class UnifiedDataset(torch.utils.data.Dataset):
            for key in self.data_file_keys:
                if key in data:
                    if key in self.special_operator_map:
-                        data[key] = self.special_operator_map[key]
+                        data[key] = self.special_operator_map[key](data[key])
                    elif key in self.data_file_keys:
                        data[key] = self.main_data_operator(data[key])
        return data