rixty/KoboldAI-Client
1
1
Fork 0
mirror of https://github.com/KoboldAI/KoboldAI-Client.git synced 2025-06-05 21:59:24 +02:00
Code Issues Projects Releases Wiki Activity

Remove unused file open

Browse Source
This commit is contained in:
somebody
2023-07-03 17:51:54 -05:00
parent 7f869a54d8
commit 32917fd651
1 changed files with 102 additions and 109 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
tpu_mtj_backend.py
View File

@@ -1171,12 +1171,7 @@ def load_model(path: str, model_type: str, badwordsids=koboldai_settings.badword
if callback.nested: if callback.nested:
return return
callback.nested = True callback.nested = True
with zipfile.ZipFile(f, "r") as z:
try: try:
last_storage_key = None
zipfolder = os.path.basename(os.path.normpath(f)).split('.')[0]
f = None
current_offset = 0
if utils.current_shard == 0: if utils.current_shard == 0:
print("\n\n\nThis model has ", f"{hk.data_structures.tree_size(network.state['params']):,d}".replace(",", " "), " parameters.\n") print("\n\n\nThis model has ", f"{hk.data_structures.tree_size(network.state['params']):,d}".replace(",", " "), " parameters.\n")
@@ -1288,8 +1283,6 @@ def load_model(path: str, model_type: str, badwordsids=koboldai_settings.badword
utils.bar = None utils.bar = None
koboldai_vars.status_message = "" koboldai_vars.status_message = ""
callback.nested = False callback.nested = False
if isinstance(f, zipfile.ZipExtFile):
f.close()
callback.nested = False callback.nested = False
if os.path.isdir(koboldai_vars.model.replace('/', '_')): if os.path.isdir(koboldai_vars.model.replace('/', '_')):