diff --git a/fishsense_lite/commands/label_studio.py b/fishsense_lite/commands/label_studio.py
index 4882a88..144267b 100644
--- a/fishsense_lite/commands/label_studio.py
+++ b/fishsense_lite/commands/label_studio.py
@@ -29,7 +29,7 @@
 from fishsense_lite.utils import get_output_file, get_root, uint16_2_uint8
 
 
-@ray.remote(vram_mb=768)
+@ray.remote(vram_mb=900)
 def execute_nn_laser(
     input_file: Path,
     lens_calibration: LensCalibration,