Dreambooth
Dreambooth copied to clipboard
brian6091
Praise: Very excited for the ti branch!
Hey @brian6091,
Awesome work! Best SD Colab notebooks out there are yours! Been playing around with them for a couple of weeks with LoRA.
Can't wait for the ti branch with config Colab to be ready :)
Let me know how I can help out.
@amerkay Glad you can follow given the rather uninformative commit messages! Lots of changes coming up. I should have it ready for testing in a couple of days. Inference part still requires some work given the number of possibilities that have to be taken into account...
I can help with that. I had to modify the main branch notebook to "monkey patch" LoRA for inference.
Still wrapping my head around the embedding resizing for the TI method. Got a bunch of errors, but that might be because I was trying to speed things up with Flax/JAX code at the same time.
No rush! Great work, mate!
Cool, I'll ping you when I get the training part finished. I'm waiting for the safetensors pull request to get merged into LoRA to decide exactly how to go about saving model bits.
@amerkay
I can help with that. I had to modify the main branch notebook to "monkey patch" LoRA for inference.
Ah yes, I forgot to do that. I've tucked away the current main branch in v0.0.1 branch. If you want to make a pull request on that branch I'd appreciate that.
Ah yes, I forgot to do that. I've tucked away the current main branch in v0.0.1 branch. If you want to make a pull request on that branch I'd appreciate that.
You lost me :) Not much going on in v0.0.1 branch:
Sorry, I just meant for the inference part that you had to add to the notebook on main
No worries - done, pull request at https://github.com/brian6091/Dreambooth/pull/18