From 008499b7252118f59fb4b311514207a5f49db753 Mon Sep 17 00:00:00 2001 From: technillogue Date: Fri, 17 May 2024 21:19:33 -0400 Subject: [PATCH] new concurrency field --- modules/cog-interface.nix | 15 +++++++++++---- 1 file changed, 11 insertions(+), 4 deletions(-) diff --git a/modules/cog-interface.nix b/modules/cog-interface.nix index a21bbe2..8f33f0c 100644 --- a/modules/cog-interface.nix +++ b/modules/cog-interface.nix @@ -46,10 +46,17 @@ in default = null; example = "train.py:train"; }; - concurrency = mkOption { - type = types.int; - default = 1; - description = "Allowed concurrency"; + concurrency = { + max = mkOption { + type = types.int; + default = 1; + description = "Allowed concurrency"; + }; + default_target = mkOption { + type = types.int; + default = 1; + description = "Target concurrency"; + }; }; }; options.cognix = with lib; {