Update run_exp.py

ignorejjj · web-flow · commit 69187858a945 · 2024-12-08T22:29:40.000+08:00
diff --git a/examples/methods/run_exp.py b/examples/methods/run_exp.py
@@ -563,6 +563,7 @@ def rqrag(args):
     config_dict = {
         "save_note": save_note,
         "gpu_id": args.gpu_id,
+        'framework': 'vllm',
         "dataset_name": args.dataset_name,
         "split": args.split,
         "retrieval_topk": 5,
