bugfix

app/models/heads/yoso_head.py

@@ -360,7 +360,8 @@ class CrossAttenHead(nn.Module):
                 torch.nn.init.orthogonal_(cls_embed)
                 cls_embed = cls_embed[:, None]
             else:
-                ov_path = os.path.join(os.path.expanduser('~/.cache/embd'), f"{ov_classifier_name}.pth")
+                # ov_path = os.path.join(os.path.expanduser('~/.cache/embd'), f"{ov_classifier_name}.pth")
+                ov_path = os.path.join('./models/', f"{ov_classifier_name}.pth")
                 cls_embed = torch.load(ov_path)
                 cls_embed_norm = cls_embed.norm(p=2, dim=-1)
                 assert torch.allclose(cls_embed_norm, torch.ones_like(cls_embed_norm))