update qwen3.5 only text conpute pos bug
This commit is contained in:
@@ -1068,6 +1068,7 @@ impl Qwen3_5Model {
|
|||||||
(seqlen_offset + seq_len) as i64,
|
(seqlen_offset + seq_len) as i64,
|
||||||
input_ids.device(),
|
input_ids.device(),
|
||||||
)?
|
)?
|
||||||
|
.to_dtype(self.rope_deltas.as_ref().unwrap().dtype())?
|
||||||
.unsqueeze(0)?
|
.unsqueeze(0)?
|
||||||
.broadcast_as((bs, seq_len))?
|
.broadcast_as((bs, seq_len))?
|
||||||
.broadcast_add(self.rope_deltas.as_ref().unwrap())?
|
.broadcast_add(self.rope_deltas.as_ref().unwrap())?
|
||||||
|
|||||||
Reference in New Issue
Block a user