aidlab/AiDA_Python
6
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

feat 2倍超分

Browse Source
This commit is contained in:
zhouchengrong
2024-04-11 16:27:40 +08:00
parent d4f8872574
commit e20afaac05
1 changed files with 1 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
app/service/super_resolution/service.py
View File

@@ -59,8 +59,7 @@ class SuperResolution:
@RunTime @RunTime
def infer(self, inputs): def infer(self, inputs):
return self.triton_client.async_infer( return self.triton_client.async_infer(
# model_name=SR_MODEL_NAME, model_name=SR_MODEL_NAME,
model_name="super_resolution_size-2",
inputs=inputs, inputs=inputs,
callback=self.callback callback=self.callback
) )