feat 2倍超分

2024-04-11 16:27:40 +08:00
parent d4f8872574
commit e20afaac05
1 changed files with 1 additions and 2 deletions
--- a/app/service/super_resolution/service.py
+++ b/app/service/super_resolution/service.py
@@ -59,8 +59,7 @@ class SuperResolution:
    @RunTime
    def infer(self, inputs):
        return self.triton_client.async_infer(
-            # model_name=SR_MODEL_NAME,
-            model_name="super_resolution_size-2",
+            model_name=SR_MODEL_NAME,
            inputs=inputs,
            callback=self.callback
        )