k_lms sampling working; half precision working, can override with --full_precision

This commit is contained in:
Lincoln Stein 2022-08-21 20:16:31 -04:00
parent 2ace56313c
commit f5450bad61

View File

@ -410,8 +410,7 @@ class T2I:
model.cuda()
model.eval()
if self.full_precision:
print('Using slower but more accurate full precision math')
model.full()
print('Using slower but more accurate full-precision math (--full_precision)')
else:
print('Using half precision math. Call with --full_precision to use full precision')
model.half()