Revert parameter change and disable segs instead

2024-05-04 18:18:22 +03:00
parent 6493792a70
commit fe4f477d79
2 changed files with 7 additions and 6 deletions
@@ -26,7 +26,8 @@ class UniFusion(nn.Module):
        self.use_seperate_tokenizer = use_seperate_tokenizer

        # Use instance masks as additional model inputs for mask conditioned image generation
-        self.use_segs = True if self.add_masks else False
+        #self.use_segs = True if self.add_masks else False
+        self.use_segs = False

        if self.use_segs:
            in_dim = 30