recitalAI / ragtime-package

Ragtime🎹 is an LLMOps framework to automate testing and comparison for text to text large language models
MIT License
9 stars 5 forks source link

Add a config file for experiments #5

Open recitalAI opened 3 months ago

recitalAI commented 3 months ago

Too many parameters are shared between the different generation steps. It would be more convenient to use a configuration file to define all the steps and run them at once.

GillesMoyse commented 3 months ago

In progress (cf. PR from Youssouf)

GillesMoyse commented 3 months ago

The configuration file describes how an Expe will go from one step to another, in which order. Thus, the configuration file is made of the 3 main blocks :

  1. an Expe file
  2. a dictionary of Steps
  3. a workflow, an ordered list of Steps

Expe file : a string to the Expe file

Each Step is defined as follows:

Each TextGenerator comprises:

Below is an example of a configuration file with 3 steps, AnswerGen, FactsGen and EvalGen:

pipeline = load_pipeline(path_to_pipeline.toml) pipeline.run()