nicolabs/Open-Sora
1
0
Fork 0
mirror of https://github.com/hpcaitech/Open-Sora.git synced 2026-05-21 11:59:01 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions 1

minor fix

Browse source
This commit is contained in:
zhengzangw 2024-06-21 19:22:02 +00:00
parent ba65d369db
commit a6bdabe286
1 changed files with 1 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
opensora/datasets/datasets.py
View file

@ -171,7 +171,7 @@ class VariableVideoTextDataset(VideoTextDataset):
# repeat
video = image.unsqueeze(0)
# # TCHW -> CTHW
# TCHW -> CTHW
video = video.permute(1, 0, 2, 3)
ret = {
"video": video,