Official Pytorch implementation for our paper CoIn: A Lightweight and Effective Framework for Story Visualization and Continuation by Ming Tao, Bing-Kun Bao, Hao Tang, Yaowei Wang, Changsheng Xu.
Generated Images
- python 3.9
- Pytorch 1.9
- At least 1x24GB 3090 GPU (for training)
- Only CPU (for sampling)
Details can be found in https://github.com/tobran/CoIn