rixty/KoboldAI-Client
1
1
Fork 0
mirror of https://github.com/KoboldAI/KoboldAI-Client.git synced 2025-01-07 13:52:10 +01:00
Code Issues Projects Releases Wiki Activity

Cleaner implementation of zipfolder

Browse Source
This commit is contained in:
Henk 2022-12-10 19:23:08 +01:00
parent 9a8e8a0005
commit eeb1774d42
2 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
aiserver.py
View File

@ -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)

2
tpu_mtj_backend.py
View File

@ -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)