address #9

lucidrains · Jan 31, 2024 · bb5c1e3 · bb5c1e3
1 parent 8ae68ff
commit bb5c1e3
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 1 deletion.
diff --git a/q_transformer/agent.py b/q_transformer/agent.py
@@ -1,3 +1,4 @@
+import sys
 from pathlib import Path
 
 from numpy.lib.format import open_memmap
@@ -18,6 +19,10 @@
 
 from tqdm import tqdm
 
+# just force training on 64 bit systems
+
+assert sys.maxsize > (2 ** 32), 'you need to be on 64 bit system to store > 2GB experience for your q-transformer agent'
+
 # constants
 
 TEXT_EMBEDS_FILENAME = 'text_embeds.memmap.npy'

diff --git a/setup.py b/setup.py
@@ -3,7 +3,7 @@
 setup(
   name = 'q-transformer',
   packages = find_packages(exclude=[]),
-  version = '0.1.9',
+  version = '0.1.10',
   license='MIT',
   description = 'Q-Transformer',
   author = 'Phil Wang',