oops

lucidrains · Nov 28, 2023 · 641c1de · 641c1de
1 parent f3b4959
commit 641c1de
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 2 deletions.
diff --git a/q_transformer/robotic_transformer.py b/q_transformer/robotic_transformer.py
@@ -811,5 +811,4 @@ def forward(
         if self.num_actions == 1:
             q_values = rearrange(q_values, '... 1 b -> ... b')
 
-        exit()
         return q_values
diff --git a/setup.py b/setup.py
@@ -3,7 +3,7 @@
 setup(
   name = 'q-transformer',
   packages = find_packages(exclude=[]),
-  version = '0.0.14',
+  version = '0.0.15',
   license='MIT',
   description = 'Q-Transformer',
   author = 'Phil Wang',