We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent a5378a8 commit a646882Copy full SHA for a646882
open_flamingo/eval/eval_models/open_flamingo.py
@@ -16,7 +16,7 @@ class EvalModel(BaseEvalModel):
16
def __init__(self, model_args):
17
super().__init__(model_args)
18
19
- if model_args["model_family"] == "openflamingo":
+ if model_args["model_family"] == "flamingo":
20
assert "cross_attn_every_n_layers" in model_args, "cross_attn_every_n_layers is required for Flamingo models"
21
else:
22
assert "cross_attn_every_n_layers" not in model_args, "cross_attn_every_n_layers is only for Flamingo models"
0 commit comments