Spaces:
Runtime error
Runtime error
Update model.py
Browse files
model.py
CHANGED
|
@@ -347,7 +347,7 @@ def get_1d_sincos_pos_embed_from_grid(embed_dim, pos):
|
|
| 347 |
out: (M, D)
|
| 348 |
"""
|
| 349 |
assert embed_dim % 2 == 0
|
| 350 |
-
omega = np.arange(embed_dim // 2, dtype=np.
|
| 351 |
omega /= embed_dim / 2.
|
| 352 |
omega = 1. / 10000**omega # (D/2,)
|
| 353 |
|
|
|
|
| 347 |
out: (M, D)
|
| 348 |
"""
|
| 349 |
assert embed_dim % 2 == 0
|
| 350 |
+
omega = np.arange(embed_dim // 2, dtype=np.float32)
|
| 351 |
omega /= embed_dim / 2.
|
| 352 |
omega = 1. / 10000**omega # (D/2,)
|
| 353 |
|