We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 9ac9d69 commit d0ca80eCopy full SHA for d0ca80e
2 files changed
enformer_pytorch/finetune.py
@@ -150,7 +150,7 @@ def __init__(
150
self.scale = dim_head ** -0.5
151
self.heads = heads
152
inner_dim = heads * dim_head
153
- self.to_queries = nn.Linear(enformer_hidden_dim, inner_dim)
+ self.to_queries = nn.Linear(enformer_hidden_dim, inner_dim, bias = False)
154
155
self.null_key = nn.Parameter(torch.randn(inner_dim))
156
self.null_value = nn.Parameter(torch.randn(inner_dim))
setup.py
@@ -4,7 +4,7 @@
4
name = 'enformer-pytorch',
5
packages = find_packages(exclude=[]),
6
include_package_data = True,
7
- version = '0.1.18',
+ version = '0.1.19',
8
license='MIT',
9
description = 'Enformer - Pytorch',
10
author = 'Phil Wang',
0 commit comments