Skip to content

Commit a646882

Browse files
Update open_flamingo.py
1 parent a5378a8 commit a646882

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

open_flamingo/eval/eval_models/open_flamingo.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -16,7 +16,7 @@ class EvalModel(BaseEvalModel):
1616
def __init__(self, model_args):
1717
super().__init__(model_args)
1818

19-
if model_args["model_family"] == "openflamingo":
19+
if model_args["model_family"] == "flamingo":
2020
assert "cross_attn_every_n_layers" in model_args, "cross_attn_every_n_layers is required for Flamingo models"
2121
else:
2222
assert "cross_attn_every_n_layers" not in model_args, "cross_attn_every_n_layers is only for Flamingo models"

0 commit comments

Comments
 (0)