/CHAE

CHAE: Fine-Grained Controllable Story Generation with Characters, Actions and Emotions

Primary LanguagePython

CHAE: Fine-Grained Controllable Story Generation with Characters, Actions and Emotions (COLING 2022)

This repository will be continuously improved.

Requirements

  • accelerate >= 0.12.0
  • datasets >= 1.8.0
  • sentencepiece != 0.1.92
  • protobuf
  • sacrebleu >= 1.4.12
  • py7zr
  • torch >= 1.3

Checkpoints

The checkpoint can be dowloaded from here.

Inference

Inference stories with test data.

sh run_inference.sh

Arguments include:

  • --device. The gpu id you want to use.
  • --model_name_or_path. The checkpoint path would be loaded.
  • --num_beams. The beam size.
  • --do_sample. Whether use sampling decoding.
  • --top_k.
  • --temperature.

Training

The training steps will be added soon.