Skip to content

Commit 4055899

Browse files
committed
default to 4 residual streams, hyper-connections paper from bytedance
1 parent 2efe221 commit 4055899

File tree

3 files changed

+4
-3
lines changed

3 files changed

+4
-3
lines changed

meshgpt_pytorch/meshgpt_pytorch.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1105,7 +1105,8 @@ def __init__(
11051105
ff_glu = True,
11061106
attn_num_mem_kv = 4,
11071107
add_value_residual = True,
1108-
learned_value_residual_mix = True
1108+
learned_value_residual_mix = True,
1109+
num_residual_streams = 4
11091110
),
11101111
cross_attn_num_mem_kv = 4, # needed for preventing nan when dropping out text condition
11111112
dropout = 0.,

meshgpt_pytorch/version.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1 +1 @@
1-
__version__ = '1.7.2'
1+
__version__ = '1.8.0'

setup.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -40,7 +40,7 @@
4040
'torch_geometric',
4141
'tqdm',
4242
'vector-quantize-pytorch>=1.18.1',
43-
'x-transformers>=1.42.24',
43+
'x-transformers>=1.43.2',
4444
],
4545
setup_requires=[
4646
'pytest-runner',

0 commit comments

Comments
 (0)