apextoaster
/
onnx-web
1
0
Fork 0
Code Actions

fix(api): correctly load server env for conversion

This commit is contained in:
Sean Sube 2023-03-19 10:39:09 -05:00
parent 98e488319c
commit b8f816323e
Signed by: ssube
GPG Key ID: 3EED7B957D362AF1
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
api/onnx_web/convert/utils.py
View File

@ -23,7 +23,7 @@ LegacyModel = Tuple[str, str, Optional[bool], Optional[bool], Optional[int]]
class ConversionContext(ServerContext):
def __init__(
self,
model_path: str,
model_path: Optional[str] = None,
cache_path: Optional[str] = None,
device: Optional[str] = None,
half: Optional[bool] = False,
@ -32,7 +32,7 @@ class ConversionContext(ServerContext):
prune: Optional[List[str]] = None,
**kwargs,
) -> None:
super().__init__(model_path=model_path, cache_path=cache_path)
super().__init__(model_path=model_path, cache_path=cache_path, **kwargs)
self.half = half
self.opset = opset