From d4f8872574e83451ae6fa635490c5e9c7cf84dff Mon Sep 17 00:00:00 2001 From: zhouchengrong Date: Thu, 11 Apr 2024 16:15:02 +0800 Subject: [PATCH] =?UTF-8?q?feat=202=E5=80=8D=E8=B6=85=E5=88=86?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- app/service/super_resolution/service.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/app/service/super_resolution/service.py b/app/service/super_resolution/service.py index 1cfd386..9fb9cac 100644 --- a/app/service/super_resolution/service.py +++ b/app/service/super_resolution/service.py @@ -59,7 +59,8 @@ class SuperResolution: @RunTime def infer(self, inputs): return self.triton_client.async_infer( - model_name=SR_MODEL_NAME, + # model_name=SR_MODEL_NAME, + model_name="super_resolution_size-2", inputs=inputs, callback=self.callback )