From b82f0d5c0c9790edf70c9b3b1626c1ddc2378b5b Mon Sep 17 00:00:00 2001 From: mrq Date: Mon, 27 May 2024 19:47:58 -0500 Subject: [PATCH] finally nailed the issue that caused logging to break on one machine but not another (bitnet includes zetascale which is a parasite that will break logging) --- vall_e/models/base.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/vall_e/models/base.py b/vall_e/models/base.py index c7964e9..321f654 100755 --- a/vall_e/models/base.py +++ b/vall_e/models/base.py @@ -56,6 +56,10 @@ except Exception as e: try: from bitnet.bit_transformer import Transformer as BitNetTransformerBlock, RMSNorm as BitNetRMSNorm + # re-enable logging because zetascale fucking sucks + import logging + logging.getLogger().setLevel(logging.DEBUG) + # override for wrapping checkpointing def BitNetTransformerBlock_forward(self, x: Tensor, *args, **kwargs) -> Tensor: skip = x