From 477ff355177124ec9a2a538aae0b535f9f72fede Mon Sep 17 00:00:00 2001 From: Won-Kyu Park Date: Thu, 5 Sep 2024 09:19:04 +0900 Subject: [PATCH] preserve detected dtype_inference --- modules/shared_init.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/modules/shared_init.py b/modules/shared_init.py index a6ad0433d..2d58c6374 100644 --- a/modules/shared_init.py +++ b/modules/shared_init.py @@ -29,7 +29,7 @@ def initialize(): devices.dtype = torch.float32 if cmd_opts.no_half else torch.float16 devices.dtype_vae = torch.float32 if cmd_opts.no_half or cmd_opts.no_half_vae else torch.float16 - devices.dtype_inference = torch.float32 if cmd_opts.precision == 'full' else devices.dtype + devices.dtype_inference = torch.float32 if cmd_opts.precision == 'full' else devices.dtype_inference if cmd_opts.precision == "half": msg = "--no-half and --no-half-vae conflict with --precision half"