eeuuia commited on
Commit
a5cd3be
·
verified ·
1 Parent(s): 81a66b1

Update utils/debug_utils.py

Browse files
Files changed (1) hide show
  1. utils/debug_utils.py +7 -5
utils/debug_utils.py CHANGED
@@ -7,15 +7,17 @@ import torch
7
 
8
  # Define o nível de log. Mude para "INFO" para desativar os logs detalhados.
9
  # Você pode controlar isso com uma variável de ambiente.
 
10
  import warnings
11
  warnings.filterwarnings("ignore", category=UserWarning)
12
  warnings.filterwarnings("ignore", category=FutureWarning)
13
  warnings.filterwarnings("ignore", message=".*")
14
- from huggingface_hub import logging
15
- logging.set_verbosity_error()
16
- logging.set_verbosity_warning()
17
- logging.set_verbosity_info()
18
- logging.set_verbosity_debug()
 
19
 
20
 
21
  def _format_value(value):
 
7
 
8
  # Define o nível de log. Mude para "INFO" para desativar os logs detalhados.
9
  # Você pode controlar isso com uma variável de ambiente.
10
+ import logging
11
  import warnings
12
  warnings.filterwarnings("ignore", category=UserWarning)
13
  warnings.filterwarnings("ignore", category=FutureWarning)
14
  warnings.filterwarnings("ignore", message=".*")
15
+ from huggingface_hub import logging as ll
16
+ ll.set_verbosity_error()
17
+ ll.set_verbosity_warning()
18
+ ll.set_verbosity_info()
19
+ ll.set_verbosity_debug()
20
+
21
 
22
 
23
  def _format_value(value):