Spaces:

X-iZhang
/

Libra

Running

X-iZhang commited on Jan 20

Commit

bf1a4c1

verified ·

1 Parent(s): bf1e548

Update libra/eval/run_libra.py

Files changed (1) hide show

libra/eval/run_libra.py CHANGED Viewed

@@ -176,7 +176,7 @@ def libra_eval(
     attention_mask = torch.ones(input_ids.shape, dtype=torch.long, device=device)
     pad_token_id = tokenizer.pad_token_id
-    image_tensor = get_image_tensors(image_file, image_processor, model, , device=device)
     stop_str = conv.sep if conv.sep_style != SeparatorStyle.TWO else conv.sep2
     keywords = [stop_str]

     attention_mask = torch.ones(input_ids.shape, dtype=torch.long, device=device)
     pad_token_id = tokenizer.pad_token_id
+    image_tensor = get_image_tensors(image_file, image_processor, model, device=device)
     stop_str = conv.sep if conv.sep_style != SeparatorStyle.TWO else conv.sep2
     keywords = [stop_str]