This commit is contained in:
Qing
2023-02-19 21:16:30 +08:00
parent d41cf9b09d
commit 49fcb53833
2 changed files with 10 additions and 19 deletions

View File

@@ -48,7 +48,8 @@ def ceil_modulo(x, mod):
return (x // mod + 1) * mod
def load_jit_model(url_or_path, device):
def \
load_jit_model(url_or_path, device):
if os.path.exists(url_or_path):
model_path = url_or_path
else: