HomeAI Tools
ChineseAiDungeon

ChineseAiDungeon

3 liked
entry-slick
entry-slick
About ChineseAiDungeon

AI Dungeon in Chinese is a text adventure game that uses GPT-2 and is fine-tuned using the Qingyuan CPM pre-training model.

How to start the game?

  • You can choose to experience the game directly on Colab
  • clone the code to your local machine and run the game using the ChineseAiDungeonColabDemo.ipynb file.

    If you choose to run the game locally, you need to download and unzip the storyTeller1.1 model beforehand to the project root directory.

Before starting the game

  • The game can run on CPU or GPU, but the model inference on CPU will be very slow and not recommended.
  • You need at least 8GB of VRAM to run this program.
  • The fine-tuning corpus is the text from the Choose Your Story website translated by Google, and due to the translation process, the quality is not good. Therefore, do not have high expectations for the stories generated by ChineseAiDungeon.
  • If the model's output doesn't respond to your actions reasonably or keeps repeating certain facts, try ignoring the model's repetition and directly input more specific actions. You will eventually progress.
  • Please ignore some of the logical issues in the model's output and focus on the general story direction for a better gaming experience.

Ask for help

You can learn about the creation process of the Chinese version of the AI dungeon through these three detailed logs: - http://icybee.cn/article/71.html - http://icybee.cn/article/72.html - http://icybee.cn/article/73.html

As you can see, the quality of the stories generated by ChineseAiDungeon is not excellent, and some algorithm or data improvements may help improve the quality of the stories. If you have any ideas, please feel free to raise an issue/pull request for this project or contact me at icybee@yeah.net.

Visit Official Website

https://github.com/bupticybee/ChineseAiDungeon