Adding paper
This commit is contained in:
@@ -6,6 +6,7 @@ Collects data across multiple prompts for research analysis.
|
|||||||
import subprocess
|
import subprocess
|
||||||
import argparse
|
import argparse
|
||||||
import json
|
import json
|
||||||
|
import sys
|
||||||
from pathlib import Path
|
from pathlib import Path
|
||||||
import time
|
import time
|
||||||
from typing import List
|
from typing import List
|
||||||
@@ -22,7 +23,7 @@ def run_benchmark(
|
|||||||
):
|
):
|
||||||
"""Run a single benchmark."""
|
"""Run a single benchmark."""
|
||||||
cmd = [
|
cmd = [
|
||||||
'python', 'inference.py',
|
sys.executable, 'inference.py',
|
||||||
'--checkpoint', checkpoint,
|
'--checkpoint', checkpoint,
|
||||||
'--prompt', prompt,
|
'--prompt', prompt,
|
||||||
'--max-length', str(max_length),
|
'--max-length', str(max_length),
|
||||||
|
|||||||
Reference in New Issue
Block a user