diff --git a/setup.py b/setup.py index d146f2c..dfa6d3d 100644 --- a/setup.py +++ b/setup.py @@ -3,7 +3,7 @@ setup( name = 'muse-maskgit-pytorch', packages = find_packages(exclude=[]), - version = '0.0.11', + version = '0.0.12', license='MIT', description = 'MUSE - Text-to-Image Generation via Masked Generative Transformers, in Pytorch', author = 'Phil Wang', @@ -23,6 +23,7 @@ 'einops>=0.6', 'ema-pytorch', 'pillow', + 'sentencepiece', 'torch>=1.6', 'transformers', 'torch>=1.6',