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

better reuse.

Browse Source
This commit is contained in:
sayakpaul
2024-11-19 10:51:00 +05:30
parent 4216705510
commit 440dc257c8
2 changed files with 19 additions and 217 deletions
Download Patch File Download Diff File Expand all files Collapse all files
training/mochi-1/text_to_video_lora.py
+2 -2
View File
@@ -56,11 +56,11 @@ from transformers import AutoTokenizer, T5EncoderModel
from args import get_args # isort:skip
import sys
sys.path.append("..")
sys.path.append(".")
from dataset import BucketSampler, VideoDatasetWithResizing, VideoDatasetWithResizeAndRectangleCrop # isort:skip
from text_encoder import compute_prompt_embeddings # isort:skip
from utils import get_gradient_norm, get_optimizer, prepare_rotary_positional_embeddings, print_memory, reset_memory # isort:skip
from utils import get_gradient_norm, get_optimizer, print_memory, reset_memory # isort:skip
logger = get_logger(__name__)