Mikubill / naifu

Train generative models with pytorch lightning
MIT License
294 stars 38 forks source link

Linking the nsfw model #7

Closed wktra closed 9 months ago

wktra commented 2 years ago

Hello, I don't see a discussion panel, so I apologize for asking this on the issues tab.

I have 2 questions:

1) I successfully used your script to output a 2GB ckpt. It works well. If I use this model as a base to train in dreambooth, does the resulting model keep the benefits from your script?

2) How do I link the nsfw model to this script? I can upload my nai-full-pruned to the colab folders through my gdrive, but I am having a hard time running it through the script. Could you help? @Mikubill