Skip to content

Commit f2d9dd8

Browse files
author
Haozhe Qi
committed
fixed a known issue
1 parent 9575db4 commit f2d9dd8

File tree

2 files changed

+1
-2
lines changed

2 files changed

+1
-2
lines changed

llava/action/llava_inference.py

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -46,7 +46,6 @@ def llava_inference(
4646
conv_template = "qwen_1_5"
4747

4848
options = mc_data['options'][0]
49-
5049
if test_type == 'base':
5150
question_type = "mc_top5_official_key"
5251
elif test_type == "direct_narration":

llava/train/train.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -989,7 +989,7 @@ def __init__(self, data_path: str, tokenizer: transformers.PreTrainedTokenizer,
989989

990990
from llava.action.generate_interval_pred import get_lookup_dict
991991

992-
self.train_triple_lookup = get_lookup_dict(os.path.join(self.EK100_anno_root, 'EPIC_100_train.csv'))
992+
self.train_triple_lookup = get_lookup_dict(os.path.join(self.EK100_anno_root, 'EPIC_100_train.csv'), self.eval_args.action_representation)
993993

994994
# Handle multiple JSON files specified in the data_path
995995
if "{" in data_path and "}" in data_path:

0 commit comments

Comments
 (0)