You hardcoded cuda - Githubissues

sipie800 / ComfyUI-PuLID-Flux-Enhanced

Apache License 2.0

79 stars 13 forks source link

You hardcoded cuda #12

Closed cchance27 closed 3 weeks ago

cchance27 commented 1 month ago

 cond = online_train(cond, device="cuda:1", step=train_step)

This breaks things on mac and others without cuda, and you forced it to :1 not everyone has a :1

sipie800 commented 1 month ago

Thanks for pointing. The default device has been assigned to input device. Although I'm not sure without an additional card the training is feasible since pulid flux take many VRAM itself. I will add device parameter in the future. But I never run big AI with MPS, I'm not sure if it is wise to support it.