alphagenome-pytorch 0.0.1__tar.gz → 0.0.3__tar.gz

{alphagenome_pytorch-0.0.1 → alphagenome_pytorch-0.0.3}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: alphagenome-pytorch
-Version: 0.0.1
+Version: 0.0.3
 Summary: AlphaGenome
 Project-URL: Homepage, https://pypi.org/project/alphagenome-pytorch/
 Project-URL: Repository, https://github.com/lucidrains/alphagenome
@@ -58,7 +58,7 @@ $ pip install alphagenome-pytorch
 
 ```python
 import torch
-from alphagenome import TransformerTower
+from alphagenome_pytorch import TransformerTower
 
 transformer = TransformerTower(dim = 768, dim_pairwise = 128)
 

{alphagenome_pytorch-0.0.1 → alphagenome_pytorch-0.0.3}/README.md

@@ -14,7 +14,7 @@ $ pip install alphagenome-pytorch
 
 ```python
 import torch
-from alphagenome import TransformerTower
+from alphagenome_pytorch import TransformerTower
 
 transformer = TransformerTower(dim = 768, dim_pairwise = 128)
 

{alphagenome_pytorch-0.0.1/alphagenome → alphagenome_pytorch-0.0.3/alphagenome_pytorch}/__init__.py

@@ -1,4 +1,4 @@
-from alphagenome.alphagenome import (
+from alphagenome_pytorch.alphagenome import (
     AlphaGenome,
     Attention,
     TransformerTower

{alphagenome_pytorch-0.0.1/alphagenome → alphagenome_pytorch-0.0.3/alphagenome_pytorch}/alphagenome.py

@@ -213,8 +213,8 @@ class SingleToPairwise(Module):
         self.split_heads = Rearrange('b n (h d) -> b n h d', h = heads)
 
         self.to_outer_sum = Sequential(
+            nn.GELU(),
             LinearNoBias(dim, dim_pairwise * 2),
-            nn.GELU()
         )
 
         self.to_qk = LinearNoBias(dim, dim_inner * 2)

{alphagenome_pytorch-0.0.1 → alphagenome_pytorch-0.0.3}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "alphagenome-pytorch"
-version = "0.0.1"
+version = "0.0.3"
 description = "AlphaGenome"
 authors = [
     { name = "Phil Wang", email = "lucidrains@gmail.com" }
@@ -58,4 +58,4 @@ dev-dependencies = []
 allow-direct-references = true
 
 [tool.hatch.build.targets.wheel]
-packages = ["alphagenome"]
+packages = ["alphagenome_pytorch"]

{alphagenome_pytorch-0.0.1 → alphagenome_pytorch-0.0.3}/tests/test_alphagenome.py

@@ -1,6 +1,6 @@
 import pytest
 import torch
-from alphagenome.alphagenome import TransformerTower
+from alphagenome_pytorch.alphagenome import TransformerTower
 
 def test_attention():