Looking for contributors?

huggingface / open-muse

Open reproduction of MUSE for fast text2image generation.

https://huggingface.co/openMUSE

Apache License 2.0

321 stars 26 forks source link

Looking for contributors? #58

Open nbardy opened 1 year ago

nbardy commented 1 year ago

👋 Hey I was looking to scale up some MUSE runs as well. Just saw this is going on. Would be happy to help out.

isamu-isozaki commented 1 year ago

@nbardy For sure! @patil-suraj might be a better person to coordinate with but there is a checklist at the end of the repo that has a bunch of tasks that he thought of! Also, we added some ideas we had to the issues so feel free to go ham.

isamu-isozaki commented 1 year ago

Let us know if you have any questions!

nbardy commented 1 year ago

Looks like the checklist is a little out of date seems @patil-suraj is already stated on T5.

This looks like a good place for me to start.

Train improved VQGANs models.

[ ] Add training component models for VQGAN (EMA, discriminator, LPIPS etc).
[ ] VGQAN training script.

nbardy commented 1 year ago

Can look into setting up some eval sets as well.

[ ] Collect different prompts for intermediate evaluations (Can reuse the prompts for dalle-mini, parti-prompts).

isamu-isozaki commented 1 year ago

@nbardy Ah, sry. Am already working on a vqgan training script which is in the pr! Let me know if you think anything is missing. And intermediate eval prompts sound awesome!