storytold/FineTrainers-Conditioning
2
0
Fork 0
mirror of https://github.com/storytold/FineTrainers-Conditioning.git synced 2026-10-27 08:47:10 +00:00
Code Issues Packages Projects Releases Wiki Activity

Support CogVideoX T2V (#165)

Browse Source 
* support cog t2v.

* generator.

* updates

* style

* fixes

* fix padding frames.

Co-authored-by: zRzRzRzRzRzRzR <Yuxuan.Zhang2104@student.xjtlu.edu.cn>

* revert changes related to generator.

* refactor a lot of things.

* accept revision and cache_dir.

* remove unused var

* refactoring fixes.

* refactor

* update

* update

---------

Co-authored-by: zRzRzRzRzRzRzR <Yuxuan.Zhang2104@student.xjtlu.edu.cn>
Co-authored-by: Aryan <aryan@huggingface.co>
This commit is contained in:
Sayak Paul
2025-01-03 10:53:16 +05:30
committed by GitHub
parent 2a1aa05191
commit b8352abf70
13 changed files with 654 additions and 55 deletions
Download Patch File Download Diff File Expand all files Collapse all files
finetrainers/utils/torch_utils.py
+1 -1
View File
@@ -29,7 +29,7 @@ def align_device_and_dtype(
return x
def expand_tensor_to_dims(tensor, ndim):
def expand_tensor_dims(tensor, ndim):
while len(tensor.shape) < ndim:
tensor = tensor.unsqueeze(-1)
return tensor