openai / consistency_models_cifar10

Consistency models trained on CIFAR-10, in JAX.
Apache License 2.0
136 stars 15 forks source link

Supporting consistency_models_cifar10 with PyTorch #6

Open LYMDLUT opened 8 months ago

LYMDLUT commented 8 months ago

I'm trying to convert it from a jax version to a pytorch version, and you're welcome to help! I have converted models ckpt for pytorch,and migrated pytorch models code from score sde pytorch! onedrive: https://1drv.ms/f/s!Avmh265yECFLbRR7rCWMnAiZDfA?e=UuJyn7 google drive: https://drive.google.com/drive/folders/1R8_G8jdiJfQSYfB8VTozGSvmMxGHGQOI?usp=sharing

The repo link is: https://github.com/LYMDLUT/consistency_models_cifar10_pytorch

Contine working on it!

REDzed0 commented 8 months ago

It seems that the link you provided is not working. Could you please send me the code? Many thanks.

SYZhang0805 commented 1 month ago

When I click the link, it shows that I do not have permission to access these files.