Skip to content

Commit

Permalink
headers added
Browse files Browse the repository at this point in the history
  • Loading branch information
Ilia Kulikov committed Sep 25, 2024
1 parent 8158fe0 commit 38de806
Show file tree
Hide file tree
Showing 3 changed files with 18 additions and 1 deletion.
5 changes: 5 additions & 0 deletions projects/self_taught_evaluator/examples/run_example.sh
Original file line number Diff line number Diff line change
@@ -1,5 +1,10 @@
#!/bin/bash

# Copyright (c) Meta Platforms, Inc. and affiliates.
#
# This source code is licensed under the MIT license found in the
# LICENSE file in the root directory of this source tree.

EXAMPLE_INPUTS=example_inputs.jsonl
MODEL_DIR=facebook/Self-taught-evaluator-llama3.1-70B # HF repo
EXAMPLE_OUTPUTS=example_outputs.jsonl
Expand Down
9 changes: 8 additions & 1 deletion projects/self_taught_evaluator/examples/run_model.py
Original file line number Diff line number Diff line change
@@ -1,8 +1,15 @@
"""
Copyright (c) Meta Platforms, Inc. and affiliates.
This source code is licensed under the MIT license found in the
LICENSE file in the root directory of this source tree.
"""

from pathlib import Path
from typing import Dict, List

from fire import Fire
from transformers import AutoTokenizer, LlamaTokenizer
from transformers import LlamaTokenizer
from utils import load_from_jsonl, parse_judgement, prepare_vllm_input, save_to_jsonl
from vllm import LLM, SamplingParams

Expand Down
5 changes: 5 additions & 0 deletions projects/self_taught_evaluator/examples/run_rewardbench.sh
Original file line number Diff line number Diff line change
@@ -1,5 +1,10 @@
#!/bin/bash

# Copyright (c) Meta Platforms, Inc. and affiliates.
#
# This source code is licensed under the MIT license found in the
# LICENSE file in the root directory of this source tree.

RB_INPUT=rewardbench_inputs.jsonl
RB_OUTPUT=rewardbench_results.jsonl
MODEL_DIR=/opt/hpcaas/.mounts/fs-0e3f1457c6d924fc0/kulikov/openeft_fs2_ckpts/wildchat_dpo_it1_public_recipe/checkpoints_hf/step_300_hf
Expand Down

0 comments on commit 38de806

Please sign in to comment.