Fixing a Funny Bug

erfanzar · Nov 27, 2023 · 45ed828 · 45ed828
1 parent 07d2274
commit 45ed828
Show file tree

Hide file tree

Showing 3 changed files with 2 additions and 3 deletions.
diff --git a/fjformer/__init__.py b/fjformer/__init__.py
@@ -50,4 +50,4 @@
     count_num_params
 )
 
-__version__ = '0.0.12'
+__version__ = '0.0.13'
diff --git a/fjformer/attention/flash_attention_tpu.py b/fjformer/attention/flash_attention_tpu.py
@@ -527,7 +527,6 @@ def lm_index_map(batch_index, head_index, q_seq_index, _, q_idx_ref, k_idx_ref):
             pl.BlockSpec(lambda *_: (0, 0, 0, 0), acc_scratch.shape),
         ]
     else:
-        assert False
         out_shape += [None, None, None]
         out_specs += [None, None, None]
 

diff --git a/setup.py b/setup.py
@@ -7,7 +7,7 @@
 
 setuptools.setup(
     name="fjformer",
-    version='0.0.12',
+    version='0.0.13',
     author="Erfan Zare Chavoshi",
     author_email="[email protected]",
     long_description=long_description,