Formatting changes.

This commit is contained in:
Marc-Andre Ferland 2022-10-22 19:02:02 -04:00
parent 026dd38480
commit 8adf965d0b

View File

@ -171,10 +171,10 @@ def load_model_ckpt():
if not thread_data.precision:
thread_data.precision = 'full' if thread_data.force_full_precision else 'autocast'
if not thread_data.unet_bs:
thread_data.unet_bs = 1
if thread_data.device == 'cpu':
thread_data.precision = 'full'