lllyasviel commited on
Commit
c73e08c
·
1 Parent(s): e1efba5
Files changed (1) hide show
  1. modules/launch_util.py +5 -0
modules/launch_util.py CHANGED
@@ -4,10 +4,15 @@ import importlib.util
4
  import subprocess
5
  import sys
6
  import re
 
7
 
8
  from functools import lru_cache
9
 
10
 
 
 
 
 
11
  python = sys.executable
12
  git = os.environ.get('GIT', "git")
13
  default_command_live = (os.environ.get('LAUNCH_LIVE_OUTPUT') == "1")
 
4
  import subprocess
5
  import sys
6
  import re
7
+ import logging
8
 
9
  from functools import lru_cache
10
 
11
 
12
+ logging.getLogger("torch.distributed.nn").setLevel(logging.ERROR) # sshh...
13
+ logging.getLogger("xformers").addFilter(lambda record: 'A matching Triton is not available' not in record.getMessage())
14
+
15
+
16
  python = sys.executable
17
  git = os.environ.get('GIT', "git")
18
  default_command_live = (os.environ.get('LAUNCH_LIVE_OUTPUT') == "1")