Skip to content

Commit e45cab6

Browse files
committed
default heinsen formulation to off for now until stability is resolved
1 parent 0a3aac8 commit e45cab6

File tree

2 files changed

+2
-2
lines changed

2 files changed

+2
-2
lines changed

gateloop_transformer/simplified_gate_loop.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -109,7 +109,7 @@ def __init__(
109109
self,
110110
dim,
111111
prenorm = True,
112-
use_heinsen = True,
112+
use_heinsen = False,
113113
use_jax_associative_scan = False,
114114
reverse = False
115115
):

setup.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -3,7 +3,7 @@
33
setup(
44
name = 'gateloop-transformer',
55
packages = find_packages(exclude=[]),
6-
version = '0.2.2',
6+
version = '0.2.3',
77
license='MIT',
88
description = 'GateLoop Transformer',
99
author = 'Phil Wang',

0 commit comments

Comments
 (0)