CoIn: A Lightweight and Effective Framework for Story Visualization and Continuation

Contextualize and Interchange during Coherent Story Visualization and Continuation

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

Requirements

• python 3.9
• Pytorch 1.9
• At least 1x24GB 3090 GPU (for training)
• Only CPU (for sampling)