From a7a28891b1e3d6c38a2ed5ff514610a70d7f7f10 Mon Sep 17 00:00:00 2001 From: songkey-win Date: Wed, 4 Jun 2025 18:24:17 +0800 Subject: [PATCH] lc --- hellomeme/models/hm_blocks.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/hellomeme/models/hm_blocks.py b/hellomeme/models/hm_blocks.py index 8c3cfd4..2c9cc0f 100644 --- a/hellomeme/models/hm_blocks.py +++ b/hellomeme/models/hm_blocks.py @@ -315,7 +315,10 @@ class SmallUnetV5(nn.Module): def get_posembed_linear(length, dim, dtype, device): - return get_1d_sincos_pos_embed_from_grid(dim, torch.arange(length), output_type='pt').to(dtype=dtype, device=device) + try: + return get_1d_sincos_pos_embed_from_grid(dim, torch.arange(length), output_type='pt').to(dtype=dtype, device=device) + except: + return get_1d_sincos_pos_embed_from_grid(dim, torch.arange(length)).to(dtype=dtype, device=device) class STKAttentionV5(nn.Module):