nodes/city96-ComfyUI_ExtraModels
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

PixArt fix float64 conversion error

Browse Source 
#48
This commit is contained in:
City
2024-05-25 18:09:56 +02:00
parent 4946b5038e
commit 6867aacb26
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
PixArt/models/PixArt.py
+1 -1
View File
@@ -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) pos_embed = get_2d_sincos_pos_embed_from_grid(embed_dim, grid)
if cls_token and extra_tokens > 0: if cls_token and extra_tokens > 0:
pos_embed = np.concatenate([np.zeros([extra_tokens, embed_dim]), pos_embed], axis=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): def get_2d_sincos_pos_embed_from_grid(embed_dim, grid):