search

openai / consistency_models

Official repo for consistency models.
MIT License
6.08k stars 411 forks source link

Add Dockerfile & Fix issue#3 #16

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.

Fix issue#3

Removed factory_kwargs from unet.py. With the fix, we can work in the Docker environment above. For the latest version of flash_attn, we do not need such keyword arguments. This problem can also be solved with pip install flash-attn==0.2.8, not pip install flash-attn.

discus0434 commented 1 year ago

To throw better pull request, I once close this.