search

kvablack / ddpo-pytorch

DDPO for finetuning diffusion models, implemented in PyTorch with LoRA support
MIT License
397 stars 41 forks source link

Gif visualization #1

Closed SnowdenLee closed 1 year ago

SnowdenLee commented 1 year ago

Hi, thanks for the reimplementation! Your Gif visualization using iceberg is super nice! Could you maybe also share the code of it? Thanks a lot!

kvablack commented 1 year ago

Sure thing, here it is: https://gist.github.com/kvablack/3e2d3cf849107f76b13d1e3cd31c0bf6

It takes as input MP4 videos, which I created by just loading each LoRA checkpoint in sequence and generating an image with the same seed across checkpoints.