Add spherical dist loss

app.py

@@ -46,6 +46,11 @@ class MakeCutouts(nn.Module):
             cutouts.append(cutout)
         return torch.cat(cutouts)
 
+def spherical_dist_loss(x, y):
+    x = F.normalize(x, dim=-1)
+    y = F.normalize(y, dim=-1)
+    return (x - y).norm(dim=-1).div(2).arcsin().pow(2).mul(2)
+    
 cc12m_model = hf_hub_download(repo_id="multimodalart/crowsonkb-v-diffusion-cc12m-1-cfg", filename="cc12m_1_cfg.pth")
 model = get_model('cc12m_1_cfg')()
 _, side_y, side_x = model.shape