upload

alignedthreeattn_model.py

@@ -10,7 +10,7 @@ import torch.nn.functional as F
 
 # align_weights = torch.load("align_weights.pth")
 from torch import nn
-from backbone import CLIPAttnNode, DiNOv2AttnNode, MAEAttnNode
+from alignedthreeattn_backbone import CLIPAttnNode, DiNOv2AttnNode, MAEAttnNode
 class ThreeAttnNodes(nn.Module):
     def __init__(self, align_weights):
         super().__init__()