/AMEX-codebase

Primary LanguagePythonApache License 2.0Apache-2.0

AMEX-codebase

The repo in based on LLaMA2-Accessory. Follow the instructions to install. Then replace the accessory folder with the one in this repo. Model checkpoint is at Huggingface.

New Experiments

screenspot experiment

android control experiment

Data Preparation

Run the following command in data_utils:

python amex_to_qa.py --level all --root-dir /path/to/AMEX

You can change the level to 'l1', 'l2', 'l3', 'all' to process different levels of data.

Finetune

Follow the instructions to finetune the model.

Inference

Follow the instructions to inference. Or run the following command in accessory:

bash infer.sh