Skip to content

Commit

Permalink
Let all model memory be offloaded on nvidia.
Browse files Browse the repository at this point in the history
  • Loading branch information
comfyanonymous committed Feb 21, 2025
1 parent f579a74 commit 41c30e9
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion comfy/model_management.py
Original file line number Diff line number Diff line change
Expand Up @@ -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:
Expand Down

0 comments on commit 41c30e9

Please sign in to comment.