add torch full deterministic mode

This commit is contained in:
Vladimir Mandic
2024-06-07 09:26:51 -04:00
parent 5574833f0d
commit db9718eee6
9 changed files with 61 additions and 51 deletions
+2 -2
View File
@@ -113,7 +113,7 @@ def process_diffusers(p: processing.StableDiffusionProcessing):
t0 = time.time()
sd_models_compile.check_deepcache(enable=True)
sd_models.move_model(shared.sd_model, devices.device)
hidiffusion.apply_hidiffusion(p, shared.sd_model_type)
hidiffusion.apply(p, shared.sd_model_type)
# if 'image' in base_args:
# base_args['image'] = set_latents(p)
if hasattr(shared.sd_model, 'tgate') and getattr(p, 'gate_step', -1) > 0:
@@ -123,7 +123,7 @@ def process_diffusers(p: processing.StableDiffusionProcessing):
output = shared.sd_model(**base_args)
if isinstance(output, dict):
output = SimpleNamespace(**output)
hidiffusion.remove_hidiffusion(p)
hidiffusion.unapply()
sd_models_compile.openvino_post_compile(op="base") # only executes on compiled vino models
sd_models_compile.check_deepcache(enable=False)
if shared.cmd_opts.profile: