This commit is contained in:
@@ -37,6 +37,13 @@ def warning_with_traceback(
|
||||
file: TextIOWrapper | None = None,
|
||||
line: str | None = None,
|
||||
) -> None:
|
||||
# Исключаем трейсбеки для transformers/huggingface warnings
|
||||
if "transformers" in filename or "TRANSFORMERS_CACHE" in str(message):
|
||||
# Простое логирование без трейсбека для ML библиотек
|
||||
print(f"Warning: {message} ({filename}:{lineno})")
|
||||
return
|
||||
|
||||
# Для всех остальных warnings показываем трейсбек
|
||||
tb = traceback.format_stack()
|
||||
tb_str = "".join(tb)
|
||||
print(f"{message} ({filename}, {lineno}): {category.__name__}\n{tb_str}")
|
||||
|
||||
Reference in New Issue
Block a user