upload

alignedthreeattn_model.py

@@ -8,11 +8,11 @@ import numpy as np
 import torch
 import torch.nn.functional as F
 
-align_weights = torch.load("align_weights.pth")
+# align_weights = torch.load("align_weights.pth")
 from torch import nn
 from backbone import CLIPAttnNode, DiNOv2AttnNode, MAEAttnNode
 class ThreeAttnNodes(nn.Module):
-    def __init__(self, align_weights=align_weights):
+    def __init__(self, align_weights):
         super().__init__()
         self.backbone1 = CLIPAttnNode()
         self.backbone2 = DiNOv2AttnNode()