Skip to content

Commit c4cace7

Browse files
committed
use latest vq library, as old one may have a bug
1 parent f1df5d6 commit c4cace7

1 file changed

Lines changed: 2 additions & 2 deletions

File tree

setup.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -3,7 +3,7 @@
33
setup(
44
name = 'discrete-key-value-bottleneck-pytorch',
55
packages = find_packages(exclude=[]),
6-
version = '0.0.6',
6+
version = '0.0.7',
77
license='MIT',
88
description = 'Discrete Key / Value Bottleneck - Pytorch',
99
author = 'Phil Wang',
@@ -19,7 +19,7 @@
1919
],
2020
install_requires=[
2121
'einops>=0.4',
22-
'vector-quantize-pytorch>=0.9.0',
22+
'vector-quantize-pytorch>=0.10.14',
2323
'torch>=1.6',
2424
],
2525
classifiers=[

0 commit comments

Comments
 (0)