openai / consistency_models

Official repo for consistency models.
MIT License
6.02k stars 409 forks source link

Add Dockerfile & Fix issue#3 #17

Closed discus0434 closed 1 year ago

discus0434 commented 1 year ago

Add Dockerfile

Since this, we are able to try Consistency Model in Docker environment with make build and make run. Also, I update README.md to add description about Docker installation.

Fix issue#3

Solved with installing flash-attn==0.2.8, not flash-attn in setup.py.