diff --git a/util/pos_embed.py b/util/pos_embed.py
index 6acf8bd3b3..32d8f82275 100644
--- a/util/pos_embed.py
+++ b/util/pos_embed.py
@@ -8,9 +8,9 @@
 # --------------------------------------------------------
 
 import numpy as np
-
 import torch
 
+
 # --------------------------------------------------------
 # 2D sine-cosine position embedding
 # References:
@@ -53,7 +53,7 @@ def get_1d_sincos_pos_embed_from_grid(embed_dim, pos):
     out: (M, D)
     """
     assert embed_dim % 2 == 0
-    omega = np.arange(embed_dim // 2, dtype=np.float)
+    omega = np.arange(embed_dim // 2, dtype=np.float32)
     omega /= embed_dim / 2.
     omega = 1. / 10000**omega  # (D/2,)