docs: add missing arg docstrings

Haidra-Org · Aug 26, 2024 · 59cd9d2 · 59cd9d2
1 parent 66adda6
commit 59cd9d2
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/horde_worker_regen/process_management/worker_entry_points.py b/horde_worker_regen/process_management/worker_entry_points.py
@@ -34,7 +34,10 @@ def start_inference_process(
         inference_semaphore (Semaphore): The semaphore to use to limit concurrent inference.
         disk_lock (Lock): The lock to use for disk access.
         aux_model_lock (Lock): The lock to use for auxiliary model downloading.
+        low_memory_mode (bool, optional): If true, the process will attempt to use less memory. Defaults to True.
         high_memory_mode (bool, optional): If true, the process will attempt to use more memory. Defaults to False.
+        very_high_memory_mode (bool, optional): If true, the process will attempt to use even more memory.
+            Defaults to False.
         amd_gpu (bool, optional): If true, the process will attempt to use AMD GPU-specific optimisations.
             Defaults to False.
     """