search

a-r-r-o-w / cogvideox-factory

Memory optimized finetuning scripts for CogVideoX using TorchAO and DeepSpeed
Apache License 2.0
177 stars 16 forks source link

add VideoDatasetWithResizeAndRectangleCrop dataset resize crop #13

Closed glide-the closed 3 days ago

glide-the commented 3 days ago

https://github.com/a-r-r-o-w/cogvideox-factory/issues/12

Too large an image will cause torch._dynamo.exc.TorchRuntimeError: Cannot convert -oo to int The VideoDatasetWithResizeAndRectangleCrop class is added here to crop the video