Small cleanups.

main
comfyanonymous 2 years ago
parent 9225465975
commit 0d7b0a4dc7

@ -589,7 +589,10 @@ def should_use_fp16(device=None, model_params=0):
if cpu_mode() or mps_mode(): if cpu_mode() or mps_mode():
return False #TODO ? return False #TODO ?
if torch.cuda.is_bf16_supported() or xpu_available: if xpu_available:
return True
if torch.cuda.is_bf16_supported():
return True return True
props = torch.cuda.get_device_properties("cuda") props = torch.cuda.get_device_properties("cuda")

Loading…
Cancel
Save