JLNTV_IPD/CogVideo
4
0
Fork 0
mirror of https://github.com/THUDM/CogVideo.git synced 2025-04-05 19:41:59 +08:00
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #271 from KihongK/typo-error

Browse Source 
fix brackets missing by mistake
This commit is contained in:
Yuxuan.Zhang 2024-09-12 16:58:23 +08:00 committed by GitHub
commit b1da0c94c0
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
sat/data_video.py
View File

@ -408,7 +408,7 @@ class SFTDataset(Dataset):
num_frames = self.max_num_frames
start = int(self.skip_frms_num)
end = int(ori_vlen - self.skip_frms_num)
indices = np.arange(start, end, max((end - start) // num_frames), 1).astype(int)
indices = np.arange(start, end, max((end - start) // num_frames, 1)).astype(int)
temp_frms = vr.get_batch(np.arange(start, end))
assert temp_frms is not None
tensor_frms = (