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

Remove debugging code from tpu_mtj_backend.py

Browse Source
This commit is contained in:
Gnome Ann 2022-05-14 12:00:44 -04:00
parent d5ab3ef5b1
commit d4e8f56789
1 changed files with 0 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
tpu_mtj_backend.py
View File

@ -1264,9 +1264,6 @@ def load_model(path: str, driver_version="tpu_driver0.1_dev20210607", hf_checkpo
error = f"{mk} {pk} could not be found in the model checkpoint" error = f"{mk} {pk} could not be found in the model checkpoint"
print("\n\nERROR: " + error, file=sys.stderr) print("\n\nERROR: " + error, file=sys.stderr)
raise RuntimeError(error) raise RuntimeError(error)
except:
import traceback
traceback.print_exc()
finally: finally:
if utils.num_shards is None or utils.current_shard >= utils.num_shards: if utils.num_shards is None or utils.current_shard >= utils.num_shards:
utils.bar.close() utils.bar.close()