address #20

Author: lucidrains

MEGABYTE_pytorch/megabyte.py

@@ -234,7 +234,7 @@ def __init__(
 
         assert len(dim) == self.stages
 
-        coarsest_dim, *_, fine_dim = dim
+        *_, fine_dim = dim
 
         self.max_seq_len = max_seq_len
 

setup.py

@@ -3,7 +3,7 @@
 setup(
   name = 'MEGABYTE-pytorch',
   packages = find_packages(),
-  version = '0.3.4',
+  version = '0.3.5',
   license='MIT',
   description = 'MEGABYTE - Pytorch',
   long_description_content_type = 'text/markdown',