diff --git a/README.md b/README.md index 2efba2d..6e7879a 100644 --- a/README.md +++ b/README.md @@ -236,7 +236,6 @@ You can use the following flags in the command line to change the configurations | -k, --k_shot | Used when you want to include k-shot examples in your prompt. Make sure that the column 'k_shot_prompt' exists in your questions_file. | | -d, --use_private_data | Use this to read from your own private data library. | | -o, --output_file | Output CSV file that will store your results. You need to pass the same number of output file paths as the number of prompt files | -| -bq, --bq_table | Name of BigQuery table to save to (e.g. eval.results). Remember to save your project_id as an environment variable BQ_PROJECT. | | -b, --num_beams | Indicates the number of beams you want to use for beam search at inference. Only available for `hf_runner`, `vllm_runner` and `api_runner`. | | -qz, --quantized | Indicate whether the model is an AWQ quantized model. Only available for `vllm_runner`. | | -p, --parallel_threads | The default no. of parallel threads is 5. Decrease this to 1 for gpt-4 to avoid the rate limit error. Parallelization support is currently only defined for OpenAI models. | diff --git a/main.py b/main.py index 103ab14..747cad9 100644 --- a/main.py +++ b/main.py @@ -15,7 +15,6 @@ parser.add_argument("-d", "--use_private_data", action="store_true") parser.add_argument("-k", "--k_shot", action="store_true") parser.add_argument("-o", "--output_file", nargs="+", type=str, required=True) - parser.add_argument("-bq", "--bq_table", type=str, default=None) parser.add_argument("-p", "--parallel_threads", type=int, default=5) parser.add_argument("-t", "--timeout_gen", type=float, default=30.0) parser.add_argument("-u", "--timeout_exec", type=float, default=10.0)