liuliu / swift-diffusion

BSD 3-Clause "New" or "Revised" License
423 stars 33 forks source link

How do i convert my ckpt file to s4nnc #16

Open DougStoker opened 1 year ago

liuliu commented 1 year ago

Right now, I use this script to: https://github.com/liuliu/swift-diffusion/blob/main/examples/dreambooth/main.swift But it generate fp32 ckpt, so I need to further convert that to fp16.

I am working on a new one that has no Python dependency: https://github.com/liuliu/swift-diffusion/blob/main/examples/pickle/main.swift

oniatsu commented 1 year ago

Thank you for the converter!

But I don't know something. What are the unet.ckpt, text_model.ckpt, and autoencoder.ckpt? Where can I find and download? ex. https://github.com/liuliu/swift-diffusion/blob/main/examples/dreambooth/main.swift#L1092

Now, I understand the target ckpt to convert is only this one