Disable urllib3 logger in prompt_tuner.py to disable aria2 warnings

2022-08-22 21:57:46 -04:00 · 2022-08-22 21:57:46 -04:00 · 07eb2b5c4f
parent a51e4f0651
commit 07eb2b5c4f
1 changed files with 3 additions and 0 deletions
--- a/prompt_tuner.py
+++ b/prompt_tuner.py
@ -34,6 +34,9 @@ from mkultra.tuning import GPTPromptTuningMixin, GPTNeoPromptTuningLM
 from mkultra.soft_prompt import SoftPrompt
 from typing import Dict, List, Optional, TextIO, Union

+import logging
+logging.getLogger("urllib3").setLevel(logging.ERROR)
+
 import breakmodel
 import torch_lazy_loader
 import utils