From a77ed63b6500cdc03c355a13a26af0abdb89c72a Mon Sep 17 00:00:00 2001 From: Keroosha Date: Fri, 14 Apr 2023 00:22:27 +0300 Subject: [PATCH] hardcode batchsize --- Keroosha.SilencerBot/Processing.fs | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/Keroosha.SilencerBot/Processing.fs b/Keroosha.SilencerBot/Processing.fs index 37c0098..2b58388 100644 --- a/Keroosha.SilencerBot/Processing.fs +++ b/Keroosha.SilencerBot/Processing.fs @@ -100,7 +100,7 @@ let processExecuting (job: UserJob, botConfig: Funogram.Types.BotConfig, config: async { Logging.logger.Information $"Processing {job.Id} job" let ctx = getContext job - let gpuFlag = if config.useGPU then ["--gpu 0"] else [] + let gpuFlag = if config.useGPU then ["--gpu 0 --batchsize 2"] else [] let args = List.collect id <| [ ["inference.py"; "--input"; ctx.savePath; "--output_dir"; config.tempSavePath]