preserve detected dtype_inference

This commit is contained in:
Won-Kyu Park 2024-09-05 09:19:04 +09:00
parent 24f2c1b9e4
commit 477ff35517
No known key found for this signature in database
GPG Key ID: 53AA79C8C9535D15

View File

@ -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"