diff --git a/selfdrive/modeld/runners/keras_runner.py b/selfdrive/modeld/runners/keras_runner.py
index 0a9eddeac7..f38902b7c7 100755
--- a/selfdrive/modeld/runners/keras_runner.py
+++ b/selfdrive/modeld/runners/keras_runner.py
@@ -41,10 +41,9 @@ if __name__ == "__main__":
   # limit gram alloc
   gpus = tf.config.experimental.list_physical_devices('GPU')
   if len(gpus) > 0:
-    tf.config.experimental.set_virtual_device_configuration(gpus[0], [tf.config.experimental.VirtualDeviceConfiguration(memory_limit=2048)])
+    tf.config.experimental.set_virtual_device_configuration(gpus[0], [tf.config.experimental.VirtualDeviceConfiguration(memory_limit=1024)])
 
   m = load_model(sys.argv[1])
   print(m, file=sys.stderr)
 
   run_loop(m)
-