Let all model memory be offloaded on nvidia.

This commit is contained in:
comfyanonymous 2025-02-21 06:32:11 -05:00
parent f579a740dd
commit 41c30e92e7

View File

@ -220,7 +220,7 @@ def is_amd():
MIN_WEIGHT_MEMORY_RATIO = 0.4
if is_nvidia():
MIN_WEIGHT_MEMORY_RATIO = 0.1
MIN_WEIGHT_MEMORY_RATIO = 0.0
ENABLE_PYTORCH_ATTENTION = False
if args.use_pytorch_cross_attention: