Spaces:
Running
Running
Muennighoff
commited on
Commit
•
b7b70f8
1
Parent(s):
7d1d0b3
Add d_model to parsing
Browse files
app.py
CHANGED
@@ -304,6 +304,8 @@ def get_emb_dim(model):
|
|
304 |
dim = config["hidden_dim"]
|
305 |
elif "hidden_size" in config:
|
306 |
dim = config["hidden_size"]
|
|
|
|
|
307 |
return dim
|
308 |
|
309 |
|
|
|
304 |
dim = config["hidden_dim"]
|
305 |
elif "hidden_size" in config:
|
306 |
dim = config["hidden_size"]
|
307 |
+
elif "d_model" in config:
|
308 |
+
dim = config["d_model"]
|
309 |
return dim
|
310 |
|
311 |
|