Using smaller pre-trained weight/model for prompt model

zideliu / StyleDrop-PyTorch

Unoffical implement for [StyleDrop](https://arxiv.org/abs/2306.00983)

MIT License

567 stars 27 forks source link

Using smaller pre-trained weight/model for prompt model #27

Open Youngwoo-git opened 6 months ago

Youngwoo-git commented 6 months ago

Hello

Thank you for the great work, and I'm trying to replicate some of the results that you mentioned, but unfortunately the size of model is too huge to run in my gpu (TITAN Xp). would there be smaller prompt model that i can use instead?

thanks in advance