From eeb1774d42ac4dae0298af56ac5cdd04fc8fcdc6 Mon Sep 17 00:00:00 2001 From: Henk Date: Sat, 10 Dec 2022 19:23:08 +0100 Subject: [PATCH] Cleaner implementation of zipfolder --- aiserver.py | 2 +- tpu_mtj_backend.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/aiserver.py b/aiserver.py index 48d34f43..629dcba3 100644 --- a/aiserver.py +++ b/aiserver.py @@ -2395,7 +2395,7 @@ def load_model(use_gpu=True, gpu_layers=None, disk_layers=None, initial_load=Fal try: f = z.open(f"archive/data/{storage_key}") except: - f = z.open(f"%s/data/{storage_key}" % zipfolder) + f = z.open(f"{zipfolder}/data/{storage_key}") current_offset = 0 if current_offset != model_dict[key].seek_offset: f.read(model_dict[key].seek_offset - current_offset) diff --git a/tpu_mtj_backend.py b/tpu_mtj_backend.py index 886552bd..4b27493e 100644 --- a/tpu_mtj_backend.py +++ b/tpu_mtj_backend.py @@ -1287,7 +1287,7 @@ def load_model(path: str, driver_version="tpu_driver0.1_dev20210607", hf_checkpo try: f = z.open(f"archive/data/{storage_key}") except: - f = z.open(f"%s/data/{storage_key}" % zipfolder) + f = z.open(f"{zipfolder}/data/{storage_key}") current_offset = 0 if current_offset != model_dict[key].seek_offset: f.read(model_dict[key].seek_offset - current_offset)