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:15:02 +08:00
parent e6f9a066ad
commit d4f8872574
1 changed files with 2 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -59,7 +59,8 @@ 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
) )