We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent b42b018 commit c7aa37cCopy full SHA for c7aa37c
neural_compressor/torch/algorithms/fp8_quant/utils/logger.py
@@ -103,9 +103,9 @@ def __new__(cls):
103
104
def get_enable_console_val(self):
105
enableConsole = os.environ.get("ENABLE_CONSOLE", "False").upper()
106
- if enableConsole not in ["TRUE", "FALSE"]:
107
- raise Exception("Env var 'ENABLE_CONSOLE' has to be true or false.")
108
- return enableConsole == "TRUE"
+ if enableConsole not in ["TRUE", "FALSE", "1", "0"]:
+ raise Exception("Env var 'ENABLE_CONSOLE' has to be 'true' or 'false' ('1' or '0' respectively).")
+ return enableConsole == "TRUE" or enableConsole == "1"
109
110
def get_log_level(self):
111
log_level_str = os.environ.get("LOG_LEVEL_HQT", os.environ.get("LOG_LEVEL_ALL"))
0 commit comments