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

lint changes

Browse Source
This commit is contained in:
Vladimir Mandic
2024-05-07 09:56:32 -04:00
parent 9ca5e3877d
commit b137f67edc
18 changed files with 31 additions and 21 deletions
Download Patch File Download Diff File Expand all files Collapse all files
modules/sd_models_compile.py
+3 -3
View File
@@ -78,9 +78,9 @@ def ipex_optimize(sd_model):
sd_model.prior_text_encoder = ipex_optimize_model(sd_model.prior_text_encoder)
t1 = time.time()
shared.log.info(f"IPEX Optimize: time={t1-t0:.2f}")
return sd_model
except Exception as e:
shared.log.warning(f"IPEX Optimize: error: {e}")
return sd_model
def nncf_compress_weights(sd_model):
@@ -132,9 +132,9 @@ def nncf_compress_weights(sd_model):
sd_model.prior_text_encoder = nncf_compress_model(sd_model.prior_text_encoder)
t1 = time.time()
shared.log.info(f"Compress Weights: time={t1-t0:.2f}")
return sd_model
except Exception as e:
shared.log.warning(f"Compress Weights: error: {e}")
return sd_model
def optimize_openvino(sd_model):
@@ -258,7 +258,7 @@ def compile_torch(sd_model):
shared.compiled_model_state = CompiledModelState()
return sd_model
elif shared.opts.cuda_compile_backend == "migraphx":
import torch_migraphx
import torch_migraphx # pylint: disable=unused-import
log_level = logging.WARNING if shared.opts.cuda_compile_verbose else logging.CRITICAL # pylint: disable=protected-access
if hasattr(torch, '_logging'):
torch._logging.set_logs(dynamo=log_level, aot=log_level, inductor=log_level) # pylint: disable=protected-access