Remove debugging code from tpu_mtj_backend.py

2022-05-14 12:00:44 -04:00 · 2022-05-14 12:00:44 -04:00 · d4e8f56789
parent d5ab3ef5b1
commit d4e8f56789
1 changed files with 0 additions and 3 deletions
--- a/tpu_mtj_backend.py
+++ b/tpu_mtj_backend.py
@ -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"
                                print("\n\nERROR:  " + error, file=sys.stderr)
                                raise RuntimeError(error)
-            except:
-                import traceback
-                traceback.print_exc()
            finally:
                if utils.num_shards is None or utils.current_shard >= utils.num_shards:
                    utils.bar.close()