add extra debug messages

This commit is contained in:
Vladimir Mandic
2023-05-14 12:26:15 -04:00
parent 85d67d6331
commit 760f5fb89a
2 changed files with 3 additions and 0 deletions
+1
View File
@@ -95,6 +95,7 @@ def test_fp16():
def set_cuda_params():
shared.log.debug('Verifying Torch settings')
if torch.cuda.is_available():
try:
torch.backends.cuda.matmul.allow_tf32 = shared.opts.cuda_allow_tf32
+2
View File
@@ -153,6 +153,7 @@ def select_checkpoint():
model_checkpoint = shared.opts.sd_model_checkpoint
checkpoint_info = checkpoint_aliases.get(model_checkpoint, None)
if checkpoint_info is not None or shared.cmd_opts.ckpt is not None:
shared.log.debug(f'Select checkpoint: {checkpoint_info.title if checkpoint_info is not None else None}')
return checkpoint_info
if len(checkpoints_list) == 0:
shared.log.error("Cannot run without a checkpoint")
@@ -162,6 +163,7 @@ def select_checkpoint():
if model_checkpoint is not None:
shared.log.warning(f"Default checkpoint not found: {model_checkpoint}")
shared.log.warning(f"Loading fallback checkpoint: {checkpoint_info.title}")
shared.log.debug(f'Select checkpoint: {checkpoint_info.title if checkpoint_info is not None else None}')
return checkpoint_info