From 6867aacb262d978f31af806a8b7d6d0914d9b845 Mon Sep 17 00:00:00 2001
From: City <125218114+city96@users.noreply.github.com>
Date: Sat, 25 May 2024 18:09:56 +0200
Subject: [PATCH] PixArt fix float64 conversion error

#48
---
 PixArt/models/PixArt.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/PixArt/models/PixArt.py b/PixArt/models/PixArt.py
index 7cdcb70..7850833 100644
--- a/PixArt/models/PixArt.py
+++ b/PixArt/models/PixArt.py
@@ -213,7 +213,7 @@ def get_2d_sincos_pos_embed(embed_dim, grid_size, cls_token=False, extra_tokens=
     pos_embed = get_2d_sincos_pos_embed_from_grid(embed_dim, grid)
     if cls_token and extra_tokens > 0:
         pos_embed = np.concatenate([np.zeros([extra_tokens, embed_dim]), pos_embed], axis=0)
-    return pos_embed
+    return pos_embed.astype(np.float32)
 
 
 def get_2d_sincos_pos_embed_from_grid(embed_dim, grid):