Colab execution does not work

Hello The colab notebook does not work. I found the problem is on "mae/utils/pos_embed.py", line 56 should be changed to: omega = np.arange(embed_dim // 2, dtype=np.float32) (as original np.float is deprecated) Best regards.