mteb_leaderboard_mtr

Runtime error

Muennighoff commited on Dec 22, 2022

Commit

b7b70f8

•

1 Parent(s): 7d1d0b3

Add d_model to parsing

Files changed (1) hide show

app.py CHANGED Viewed

@@ -304,6 +304,8 @@ def get_emb_dim(model):
             dim = config["hidden_dim"]
         elif "hidden_size" in config:
             dim = config["hidden_size"]
     return dim

             dim = config["hidden_dim"]
         elif "hidden_size" in config:
             dim = config["hidden_size"]
+        elif "d_model" in config:
+            dim = config["d_model"]
     return dim