Skip to content

Commit 75d2f37

Browse files
committed
fix overloading config when loading pretrained
1 parent c1012a6 commit 75d2f37

2 files changed

Lines changed: 2 additions & 2 deletions

File tree

enformer_pytorch/model_loader.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -62,7 +62,7 @@ def load_pretrained_model(
6262
params = {**config['params'], **override_params}
6363

6464
if not exists(model):
65-
model = Enformer(**config['params'])
65+
model = Enformer(**params)
6666
else:
6767
assert len(kwargs) == 0, 'you are trying to override enformer parameters, but you are already passing a reference to an instantiated enformer model'
6868

setup.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -4,7 +4,7 @@
44
name = 'enformer-pytorch',
55
packages = find_packages(exclude=[]),
66
include_package_data = True,
7-
version = '0.2.12',
7+
version = '0.2.14',
88
license='MIT',
99
description = 'Enformer - Pytorch',
1010
author = 'Phil Wang',

0 commit comments

Comments
 (0)