## Code implementation of EpicEvo

### Env setup
Run the requirements.txt using pip to install necessary packages.

### Dataset setup
1. Download the flintstones and pororo dataset.
2. Merge the downloaded content with existing data folders.
3. Process the data and save the train, val, test split as hdf5 files.

### Pretraining
1. Pretrain the story generation model using the main_eval.py, change the config file to achieve this.
2. Follow the pretraining setup of AR-LDM but train at 256x256 for efficiency.

### Customization
1. Upon having pretrained model, perform customization using main.py.
2. Configure the config file to use correct paths, hyperparameters.
3. Use the simple discriminator is enough, larger discriminator can lead to model collapse.

