remove warning

This commit is contained in:
Vladimir Mandic
2023-11-02 14:49:49 -04:00
parent 69a025f4d2
commit 5debfd2f09
+1
View File
@@ -21,6 +21,7 @@ try:
except Exception:
pass
urllib3.disable_warnings(urllib3.exceptions.InsecureRequestWarning)
warnings.filterwarnings(action="ignore", category=UserWarning, module="torchvision")
import torchvision # pylint: disable=W0611,C0411
import pytorch_lightning # pytorch_lightning should be imported after torch, but it re-enables warnings on import so import once to disable them # pylint: disable=W0611,C0411
logging.getLogger("xformers").addFilter(lambda record: 'A matching Triton is not available' not in record.getMessage())