consistency_models
consistency_models copied to clipboard
Kinyugo
A mini-library for training consistency models.
I have been trying to train and sample on all three datasets using the notebooks with the given parameters but they don't seem to generate anything but noise. Can you...
I have made this small modification by making two small changes that may be affecting performance. First, I have made sure that the same seed is being used before the...
Hi, nice repo, really appreciate it. One thing is that in the implementation of Song's consistency models, before inputting sigmas in to the network, there is a rescale: ```rescaled_t =...
Hi @Kinyugo ! Thank you so much for your great work! I had a question about this implementation. Are you able to reproduce the improved performances using the Consistency Training...
Hi @Kinyugo , Great repo, thank you for the work! I just wanted to clarify something. From my understanding, what you have implemented here is Consistency Distillation (CD), right? The...
In the class ConsistencySamplingAndEditing, the `__mask_transform` method, which I think corresponds to Algorithm 4 in the paper, essentially applies the A matrix as in the paper. In that case, the...
Nice implementation! Could you please share the results and performances of your implementation on some popular datasets? For example, the total training time, the hardware requirement, and the FID in...
Hi! Has anyone managed to reproduce the results from "Improved Techniques For Consistency Training" on the CIFAR10 dataset? Thank you for the great repository!
Excellent work! Does the author need to train the image restoration model by himself? Is there a trained one that can be tested directly?
