alphagenome-pytorch 0.0.5__tar.gz → 0.0.6__tar.gz

{alphagenome_pytorch-0.0.5 → alphagenome_pytorch-0.0.6}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: alphagenome-pytorch
-Version: 0.0.5
+Version: 0.0.6
 Summary: AlphaGenome
 Project-URL: Homepage, https://pypi.org/project/alphagenome-pytorch/
 Project-URL: Repository, https://github.com/lucidrains/alphagenome

{alphagenome_pytorch-0.0.5 → alphagenome_pytorch-0.0.6}/alphagenome_pytorch/alphagenome.py

@@ -267,6 +267,8 @@ class SingleToPairwise(Module):
 
         single = self.avg_pool(single)
 
+        pool_seq_len = single.shape[1]
+
         q, k = self.to_qk(single).chunk(2, dim = -1)
         q, k = tuple(self.split_heads(t) for t in (q, k))
 
@@ -281,10 +283,7 @@ class SingleToPairwise(Module):
             rel_q = relative_shift(einsum(q + q_rel_bias, rel_pos_encoding, 'b n h d, p h d -> b n p h'))
             rel_k = relative_shift(einsum(k + k_rel_bias, rel_pos_encoding, 'b n h d, p h d -> b n p h'))
 
-            _, seq, rel_pos_seq, _ = rel_q.shape
-            crop_padding = (rel_pos_seq - seq) // 2
-
-            rel_q, rel_k = tuple(t[..., crop_padding:(crop_padding + seq), :] for t in (rel_q, rel_k))
+            rel_q, rel_k = tuple(t[..., :pool_seq_len, :] for t in (rel_q, rel_k))
 
             rel_sim = add('b i j d, b j i d -> b i j d', rel_q, rel_k) * 0.5
 

{alphagenome_pytorch-0.0.5 → alphagenome_pytorch-0.0.6}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "alphagenome-pytorch"
-version = "0.0.5"
+version = "0.0.6"
 description = "AlphaGenome"
 authors = [
     { name = "Phil Wang", email = "lucidrains@gmail.com" }