upload models

2025-06-17 16:58:59 +08:00
parent 458be6d5bf
commit 6bdf6f0448
5 changed files with 92 additions and 85 deletions
@@ -445,8 +445,8 @@ class CopyWeights(object):
        config = dict(unet.config)

        # Need this for backwards compatibility with UNet2DConditionModel checkpoints
-        if not config.get("num_attention_heads"):
-            config["num_attention_heads"] = config["attention_head_dim"]
+        # if not config.get("num_attention_heads"):
+        #     config["num_attention_heads"] = config["attention_head_dim"]

        config = FrozenDict(config)
        model = cls.from_config(config)