rixty/KoboldAI-Client
1
1
Fork 0
mirror of https://github.com/KoboldAI/KoboldAI-Client.git synced 2024-12-11 08:16:18 +01:00
Code Issues Projects Releases Wiki Activity

Disable urllib3 logger in prompt_tuner.py to disable aria2 warnings

Browse Source
This commit is contained in:
vfbd 2022-08-22 21:57:46 -04:00
parent a51e4f0651
commit 07eb2b5c4f
1 changed files with 3 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
prompt_tuner.py
View File

@ -34,6 +34,9 @@ from mkultra.tuning import GPTPromptTuningMixin, GPTNeoPromptTuningLM
from mkultra.soft_prompt import SoftPrompt from mkultra.soft_prompt import SoftPrompt
from typing import Dict, List, Optional, TextIO, Union from typing import Dict, List, Optional, TextIO, Union
import logging
logging.getLogger("urllib3").setLevel(logging.ERROR)
import breakmodel import breakmodel
import torch_lazy_loader import torch_lazy_loader
import utils import utils