Skip to content

Commit

Permalink
removed inference settings and removed main
Browse files Browse the repository at this point in the history
  • Loading branch information
anthdr committed Jan 17, 2025
1 parent 68a0e6b commit 484b745
Showing 1 changed file with 0 additions and 10 deletions.
10 changes: 0 additions & 10 deletions eole/bin/tools/LM_scoring.py
Original file line number Diff line number Diff line change
Expand Up @@ -22,9 +22,6 @@
Below is an example of settings of a config.yaml file
verbose: false
n_best: 3
top_p: 0.9
beam_size: 10
world_size: 1
gpu_ranks: [0]
# use symlinks to last saved step
Expand Down Expand Up @@ -126,10 +123,3 @@ def run(cls, args):
ppl_file.close()

os.system('paste "' + config.src + '" "' + config.output + '".ppl > "' + config.output + '"')


if __name__ == "__main__":
parser = ArgumentParser()
LMScoring.add_args(parser)
args = parser.parse_args()
LMScoring.run(args)

0 comments on commit 484b745

Please sign in to comment.