tckm.yan/aha
0
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

update format

Browse Source
This commit is contained in:
jhqxxx
2025-10-28 14:31:02 +08:00
parent cff51418c7
commit ff5f8f87e1
2 changed files with 7 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/models/qwen2_5vl/processor.rs
+3 -2
View File
@@ -111,8 +111,9 @@ impl Qwen2_5VLProcessor {
// img_tensor: (t, c, h, w)
let t = img_tensor.dim(0)?;
let img_tensor = if t % self.vision_setting.temporal_patch_size != 0 {
let repeat_num = self.vision_setting.temporal_patch_size - t % self.vision_setting.temporal_patch_size;
let repeats = img_tensor.i(t-1)?.repeat((repeat_num, 1, 1, 1))?;
let repeat_num = self.vision_setting.temporal_patch_size
- t % self.vision_setting.temporal_patch_size;
let repeats = img_tensor.i(t - 1)?.repeat((repeat_num, 1, 1, 1))?;
Tensor::cat(&[img_tensor, &repeats], 0)?
} else {
img_tensor.clone()
src/models/qwen3vl/processor.rs
+3 -2
View File
@@ -162,8 +162,9 @@ impl Qwen3VLProcessor {
// img_tensor: (t, c, h, w)
let t = img_tensor.dim(0)?;
let img_tensor = if t % self.img_process_cfg.temporal_patch_size != 0 {
let repeat_num = self.img_process_cfg.temporal_patch_size - t % self.img_process_cfg.temporal_patch_size;
let repeats = img_tensor.i(t-1)?.repeat((repeat_num, 1, 1, 1))?;
let repeat_num = self.img_process_cfg.temporal_patch_size
- t % self.img_process_cfg.temporal_patch_size;
let repeats = img_tensor.i(t - 1)?.repeat((repeat_num, 1, 1, 1))?;
Tensor::cat(&[img_tensor, &repeats], 0)?
} else {
img_tensor.clone()