Skip to content

Commit ab51226

Browse files
committed
refactor: Update dtype in embedding_manager.py to match proj.weight
1 parent 4267c84 commit ab51226

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

examples/research_projects/anytext/embedding_manager.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -65,7 +65,7 @@ def encode_text(self, text_info):
6565

6666
if len(gline_list) > 0:
6767
recog_emb = self.get_recog_emb(gline_list)
68-
enc_glyph = self.proj(recog_emb.reshape(recog_emb.shape[0], -1).to(self.proj.weight.device))
68+
enc_glyph = self.proj(recog_emb.reshape(recog_emb.shape[0], -1).to(self.proj.weight.dtype))
6969

7070
self.text_embs_all = []
7171
n_idx = 0

0 commit comments

Comments
 (0)