storytold/FineTrainers-Conditioning
2
0
Fork 0
mirror of https://github.com/storytold/FineTrainers-Conditioning.git synced 2026-10-28 08:57: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/__init__.py
+9 -1
View File
@@ -1,4 +1,12 @@
from .diffusion_utils import default_flow_shift, resolution_dependant_timestep_flow_shift
from .diffusion_utils import (
default_flow_shift,
get_scheduler_alphas,
get_scheduler_sigmas,
prepare_loss_weights,
prepare_sigmas,
prepare_target,
resolution_dependant_timestep_flow_shift,
)
from .file_utils import delete_files, find_files
from .memory_utils import bytes_to_gigabytes, free_memory, get_memory_statistics, make_contiguous
from .optimizer_utils import get_optimizer, gradient_norm, max_gradient