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

cleanup

Browse Source 
Signed-off-by: Vladimir Mandic <mandic00@live.com>
This commit is contained in:
Vladimir Mandic
2024-09-29 20:28:01 -04:00
parent 47755dce6b
commit b31d02ba1d
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
modules/devices.py
+2 -2
View File
@@ -386,13 +386,13 @@ def set_dtype():
dtype_unet = torch.float16
if opts.no_half:
log.info('Torch override dtype: no-half set')
dtype = torch.float32
dtype_vae = torch.float32
dtype_unet = torch.float32
log.info(f'Torch override: no-half dtype={dtype}')
if opts.no_half_vae:
log.info('Torch override VAE dtype: no-half set')
dtype_vae = torch.float32
log.info(f'Torch override: no-half-vae dtype={dtype_vae}')
unet_needs_upcast = opts.upcast_sampling
if opts.inference_mode == 'inference-mode':
inference_context = torch.inference_mode