Device detection bug
Closed this issue · 2 comments
109km commented
#device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
// You should change the line above to the lines below, cause there could be "mps" device or maybe others.
from comfy import model_management
device = model_management.get_torch_device()