Public/automatic
1
0
Fork 0
mirror of https://github.com/vladmandic/automatic synced 2026-09-19 17:24:32 +02:00
Code Issues Packages Projects Releases Wiki Activity

merge: modules/loader.py

Browse Source
This commit is contained in:
vladmandic
2026-03-13 12:13:33 +01:00
parent 4bd6e42c62
commit 278c2a1b6b
1 changed files with 9 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files
modules/loader.py
+9 -6
View File
@@ -98,11 +98,16 @@ warnings.filterwarnings(action="ignore", category=UserWarning, module="torchvisi
warnings.filterwarnings(action="ignore", message="numpy.dtype size changed")
try:
import torch._logging # pylint: disable=ungrouped-imports
torch._logging._internal.DEFAULT_LOG_LEVEL = logging.ERROR # pylint: disable=protected-access
torch._logging.set_logs(all=logging.ERROR, bytecode=False, aot_graphs=False, aot_joint_graph=False, ddp_graphs=False, graph=False, graph_code=False, graph_breaks=False, graph_sizes=False, guards=False, recompiles=False, recompiles_verbose=False, trace_source=False, trace_call=False, trace_bytecode=False, output_code=False, kernel_code=False, schedule=False, perf_hints=False, post_grad_graphs=False, onnx_diagnostics=False, fusion=False, overlap=False, export=None, modules=None, cudagraphs=False, sym_node=False, compiled_autograd_verbose=False) # pylint: disable=protected-access
_compile_debug = os.environ.get('SD_COMPILE_DEBUG', None) is not None
if _compile_debug:
torch._logging._internal.DEFAULT_LOG_LEVEL = logging.ERROR # pylint: disable=protected-access
torch._logging.set_logs(dynamo=logging.WARNING, aot=logging.WARNING, inductor=logging.WARNING) # pylint: disable=protected-access
else:
torch._logging._internal.DEFAULT_LOG_LEVEL = logging.ERROR # pylint: disable=protected-access
torch._logging.set_logs(all=logging.ERROR, bytecode=False, aot_graphs=False, aot_joint_graph=False, ddp_graphs=False, graph=False, graph_code=False, graph_breaks=False, graph_sizes=False, guards=False, recompiles=False, recompiles_verbose=False, trace_source=False, trace_call=False, trace_bytecode=False, output_code=False, kernel_code=False, schedule=False, perf_hints=False, post_grad_graphs=False, onnx_diagnostics=False, fusion=False, overlap=False, export=None, modules=None, cudagraphs=False, sym_node=False, compiled_autograd_verbose=False) # pylint: disable=protected-access
import torch._dynamo
torch._dynamo.config.verbose = False # pylint: disable=protected-access
torch._dynamo.config.suppress_errors = True # pylint: disable=protected-access
torch._dynamo.config.verbose = _compile_debug # pylint: disable=protected-access
torch._dynamo.config.suppress_errors = not _compile_debug # pylint: disable=protected-access
except Exception as e:
log.warning(f'Torch logging: {e}')
if ".dev" in torch.__version__ or "+git" in torch.__version__:
@@ -119,8 +124,6 @@ timer.startup.record("bnb")
import huggingface_hub # pylint: disable=W0611,C0411
logging.getLogger("huggingface_hub.file_download").setLevel(logging.ERROR)
if huggingface_hub.__version__.startswith('0.'):
huggingface_hub.is_offline_mode = lambda: False
timer.startup.record("hfhub")
import accelerate # pylint: disable=W0611,C0411