how to select demonstration data and time-consuming problem during generating and apply multi-gpu on ditto

[yuliangCarmelo]

Hi, Omar, I found the paper truly engaging, and I'm thrilled to see that we're able to achieve such impressive personalization with just a few demonstration samples!
I am currently integrating Ditto into my application scenario and would like to ask for advice on how to select demonstration data. Additionally, I have noticed that the data generation phase is quite time-consuming and can only be performed using a single GPU, which limits the amount of training data. Are there any suggestions for improvement?
Looking forward to your reply, thank you!