search

Kinyugo / consistency_models

A mini-library for training consistency models.
https://arxiv.org/abs/2303.01469
MIT License
189 stars 20 forks source link

Colab notebook doesn't work: Import error #1

Closed vdyma closed 1 year ago

vdyma commented 1 year ago

Hi! Thanks for your work, however the Colab notebook doesn't work. I ran all the cells in order, but got an error: ModuleNotFoundError: No module named 'consistency_models'

Kinyugo commented 1 year ago

Hello,

Try restarting the notebook after installing consistency_models should fix the error.

vdyma commented 1 year ago

Thanks, it did work. I have some other ideas on improving the notebook. I will create a PR later.