search

ToTheBeginning / PuLID

[NeurIPS 2024] Official code for PuLID: Pure and Lightning ID Customization via Contrastive Alignment
Apache License 2.0
2.62k stars 184 forks source link

multi cards support？ #88

Open litt1eXixi opened 1 month ago

litt1eXixi commented 1 month ago

 Now all the weights are loaded into one card. Can we load the weights into multiple cards to avoid out of memory errors

ToTheBeginning commented 1 month ago

Yes, you can refer to https://github.com/ToTheBeginning/PuLID/issues/75#issuecomment-2359886134 and hack the codes.