tennant commited on
Commit
e2ba963
1 Parent(s): 49966bc

Update model.py

Browse files
Files changed (1) hide show
  1. model.py +1 -1
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.float)
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