From 205c64f1ea161ec2d0b3929efb73db429fa0f798 Mon Sep 17 00:00:00 2001 From: Henk Date: Sat, 13 May 2023 20:26:55 +0200 Subject: [PATCH] More universal pytorch folder detection --- modeling/inference_models/hf_torch.py | 7 +++---- 1 file changed, 3 insertions(+), 4 deletions(-) diff --git a/modeling/inference_models/hf_torch.py b/modeling/inference_models/hf_torch.py index 3f7c3967..cc7af713 100644 --- a/modeling/inference_models/hf_torch.py +++ b/modeling/inference_models/hf_torch.py @@ -581,10 +581,9 @@ class HFTorchInferenceModel(HFInferenceModel): last_storage_key = storage_key if isinstance(f, zipfile.ZipExtFile): f.close() - try: - f = z.open(f"archive/data/{storage_key}") - except: - f = z.open(f"{zipfolder}/data/{storage_key}") + ziproot = z.namelist()[0].split("/")[0] + f = z.open(f"{ziproot}/data/{storage_key}") + current_offset = 0 if current_offset != model_dict[key].seek_offset: f.read(model_dict[key].seek_offset - current_offset)