azaion/annotations
1
0
Fork 0
mirror of https://github.com/azaion/annotations.git synced 2026-04-23 05:06:31 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix converting model

Browse Source
This commit is contained in:
Oleksandr Bezdieniezhnykh
2025-09-03 14:10:46 +03:00
parent 7bbdb61253
commit a7a99c49c0
2 changed files with 3 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
Azaion.Inference/inference.pyx
+2 -1
View File
@@ -94,7 +94,8 @@ cdef class Inference:
self.ai_availability_status.set_status(AIAvailabilityEnum.ERROR, <str>str(e)) self.ai_availability_status.set_status(AIAvailabilityEnum.ERROR, <str>str(e))
onnx_engine_bytes = self.get_onnx_engine_bytes() onnx_engine_bytes = self.get_onnx_engine_bytes()
self.ai_availability_status.set_status(AIAvailabilityEnum.CONVERTING) self.ai_availability_status.set_status(AIAvailabilityEnum.CONVERTING)
model_bytes = TensorRTEngine.convert_from_onnx(res.data) model_bytes = TensorRTEngine.convert_from_onnx(onnx_engine_bytes)
self.engine = TensorRTEngine(model_bytes)
self.ai_availability_status.set_status(AIAvailabilityEnum.UPLOADING) self.ai_availability_status.set_status(AIAvailabilityEnum.UPLOADING)
res = self.loader_client.upload_big_small_resource(model_bytes, <str> engine_filename, models_dir) res = self.loader_client.upload_big_small_resource(model_bytes, <str> engine_filename, models_dir)
if res.err is not None: if res.err is not None:
Azaion.Suite.sln
+1
View File
@@ -26,6 +26,7 @@ Project("{2150E333-8FDC-42A3-9474-1A3956D46DE8}") = "Build", "Build", "{CF141A48
build\publish.cmd = build\publish.cmd build\publish.cmd = build\publish.cmd
build\requirements.txt = build\requirements.txt build\requirements.txt = build\requirements.txt
build\upload.cmd = build\upload.cmd build\upload.cmd = build\upload.cmd
build\build_local_loader_inference.cmd = build\build_local_loader_inference.cmd
EndProjectSection EndProjectSection
EndProject EndProject
Project("{FAE04EC0-301F-11D3-BF4B-00C04F79EFBC}") = "Azaion.LoaderUI", "Azaion.LoaderUI\Azaion.LoaderUI.csproj", "{C96C142E-3ED3-4455-9C22-93A12022B8A9}" Project("{FAE04EC0-301F-11D3-BF4B-00C04F79EFBC}") = "Azaion.LoaderUI", "Azaion.LoaderUI\Azaion.LoaderUI.csproj", "{C96C142E-3ED3-4455-9C22-93A12022B8A9}"