feat 2倍超分

2024-04-11 16:15:02 +08:00
parent e6f9a066ad
commit d4f8872574
1 changed files with 2 additions and 1 deletions
--- 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
        )