Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Converter from bin to ckpt works!!!!! heres link with code from Josh to incorporate to colab !!! #53

Closed
1blackbar opened this issue Oct 2, 2022 · 9 comments

Comments

@1blackbar
Copy link

1blackbar commented Oct 2, 2022

Developer named Josh did converter code in 2 days, works great i can use ckpt files in webui, does 4.3 gb file in like half minute time
You can paste the code into a cell and then paste your paths to model and ckpt location you want it in , its on bottom of the code.

https://gist.github.com/jachiam/ef621e765c2d88f9dc090ee48f6ced89

Amazing day for SD!!!

It would be nice to automate this in dreambooth notebook after training is done.

My best training so far - 10 images ( 4 body shots , 6 head shots) 1000 steps, no init word, just use class male or female, works best with likeness so far, yeh wrecks other stuff and all guys look like person trained but you can just swap models in automatics webui

@TheLastBen
Copy link
Owner

Yes just saw it, I'll add it to the colab

@TheLastBen
Copy link
Owner

if you have colab pro, let me know if the V100 and the A100 work without issues

@TheLastBen
Copy link
Owner

I'll add to the dreambooth the prior preservation to avoid messing up all the classes and get better results.

@1blackbar
Copy link
Author

Awesome! im super excited , i do have colab pro but i rarely get v100 and just once per month a100, mostly p100 not even T4!!!
But hey its on for many hours + i have 10 gmails for free colabs , i will let you know if ill get one of those

@1blackbar
Copy link
Author

1blackbar commented Oct 2, 2022

``You think You can prune the ckpt so its 2gb ?
Heres the prune code
https://github.com/JoePenna/Dreambooth-Stable-Diffusion/blob/main/prune_ckpt.py
Converts 12gb to 2gb after dreambooth on vast or runpod

# This version should automatically prune around 10GB from the ckpt file last_checkpoint_file = directory_paths[-1] + "/checkpoints/last.ckpt" !python "prune_ckpt.py" --ckpt {last_checkpoint_file}

@1blackbar
Copy link
Author

ok pruning is in the new code now by default ! Amazing

@TheLastBen
Copy link
Owner

The CKPT produced by this colab is 3.97 Gb, no need to prune

@jachiam
Copy link

jachiam commented Oct 3, 2022

Would also be grateful for attribution, @TheLastBen - if you could just put a link in the converttosd.py file that goes to my github that would be excellent. :)

@TheLastBen
Copy link
Owner

Sure, I was planning on adding that on the main Readme, thanks for the script, amazing job !

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

3 participants