| import os | |
| import argparse | |
| import json | |
| from elm.infer_elm import generate_elm_responses | |
| parser = argparse.ArgumentParser(description='run prompts with elm model.') | |
| parser.add_argument('elm_model_path', help='Path to the elm_model_path') | |
| def get_prompt_config_file(elm_model_path): | |
| return os.path.join(elm_model_path, "example_prompts.json") | |
| def run(elm_model_path: str): | |
| prompt_config_file = get_prompt_config_file(elm_model_path) | |
| with open(prompt_config_file, "r") as f: | |
| prompt_info = json.load(f) | |
| prompts = [prompt_info["template"].format(input=input) for input in prompt_info["inputs"]] | |
| print(f"Loaded prompts from: {prompt_config_file}") | |
| generate_elm_responses(elm_model_path, prompts, verbose=True) | |
| if __name__ == "__main__": | |
| args = parser.parse_args() | |
| run(args.elm_model_path) |