Batch-dim on input_ids for _embed_rows ([1, seq])
This commit is contained in:
parent
0903b849b2
commit
9fbd6fe77e
1 changed files with 1 additions and 1 deletions
|
|
@ -473,7 +473,7 @@ def build_fl2va_presentation(
|
|||
token_ids.append(int(tok))
|
||||
|
||||
seq_len = len(token_ids)
|
||||
input_ids = torch.tensor([int(t) for t in token_ids], dtype=torch.long, device=device)
|
||||
input_ids = torch.tensor([int(t) for t in token_ids], dtype=torch.long, device=device).unsqueeze(0) # [1, seq]
|
||||
|
||||
# Build the Qwen hidden-state input: scaled token embeds, with the merged vision
|
||||
# rows spliced over the pad positions.
|
||||
|
|
|
|||
Loading…
Add table
Reference in a new issue