Merge branch 'main' into feat/onnx

This commit is contained in:
Kent Keirsey
2023-07-31 16:56:34 -04:00
committed by GitHub
10 changed files with 204 additions and 67 deletions

View File

@ -276,7 +276,7 @@ class ModelInfo:
hash: str
location: Union[Path, str]
precision: torch.dtype
_cache: ModelCache = None
_cache: Optional[ModelCache] = None
def __enter__(self):
return self.context.__enter__()