README.md

November 18, 2024 ยท View on GitHub

Supported Models and Datasets

This code currently supports models and datasets hosted on Hugging Face.

Getting Started

Refer to the run.sh script for an example run.

Additional Details

  • Custom Prompts: To use specific prompts for feature extraction, refer to helpers/prompt_dict.py.
  • Adding Models: Add new models to the model_base_dict dictionary in constants.py.