Skip to content

Commit

Permalink
Revert error level changes
Browse files Browse the repository at this point in the history
  • Loading branch information
RenNagasaki committed Feb 26, 2025
1 parent dac9384 commit 990d9b2
Showing 1 changed file with 8 additions and 8 deletions.
16 changes: 8 additions & 8 deletions finetune.py
Original file line number Diff line number Diff line change
Expand Up @@ -273,21 +273,21 @@ def debug_print(
elif is_info:
prefix += ""

if level == "GPU_MEMORY" and (DebugLevels.GPU_MEMORY or is_error):
if level == "GPU_MEMORY" and DebugLevels.GPU_MEMORY:
print(f"{prefix} [GPU] {message}")
elif level == "MODEL_OPS" and (DebugLevels.MODEL_OPS or is_error):
elif level == "MODEL_OPS" and DebugLevels.MODEL_OPS:
print(f"{prefix} [MODEL] {message}")
elif level == "DATA_PROCESS" and (DebugLevels.DATA_PROCESS or is_error):
elif level == "DATA_PROCESS" and DebugLevels.DATA_PROCESS:
print(f"{prefix} [DATA] {message}")
elif level == "GENERAL" and (DebugLevels.GENERAL or is_error):
elif level == "GENERAL" and DebugLevels.GENERAL:
print(f"{prefix} [INFO] {message}")
elif level == "AUDIO" and (DebugLevels.AUDIO or is_error):
elif level == "AUDIO" and DebugLevels.AUDIO:
print(f"{prefix} [AUDIO] {message}")
elif level == "SEGMENTS" and (DebugLevels.SEGMENTS or is_error):
elif level == "SEGMENTS" and DebugLevels.SEGMENTS:
print(f"{prefix} [SEG] {message}")
elif level == "DUPLICATES" and (DebugLevels.DUPLICATES or is_error):
elif level == "DUPLICATES" and DebugLevels.DUPLICATES:
print(f"{prefix} [DUP] {message}")
elif level == "VALIDATION" and (DebugLevels.VALIDATION or is_error):
elif level == "VALIDATION" and DebugLevels.VALIDATION:
print(f"{prefix} [VAL] {message}")


Expand Down

0 comments on commit 990d9b2

Please sign in to comment.