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

Reducing cogload on the framework.

Browse Source
This commit is contained in:
CrossProduct
2025-01-24 22:46:32 +00:00
parent 305a4a994d
commit c6490c0d51
6 changed files with 158 additions and 92 deletions
Download Patch File Download Diff File Expand all files Collapse all files
finetrainers/conditioning/conditioned_pipeline.py
+4
View File
@@ -13,6 +13,10 @@ from typing import Callable, Any
import numpy as np
from diffusers.utils import is_torch_xla_available
from diffusers.pipelines.ltx.pipeline_output import LTXPipelineOutput
from diffusers.utils import is_torch_xla_available
from torch import torch
from transformers import T5EncoderModel, T5TokenizerFast
from typing import Any, Callable, Dict, List, Optional, Union
if is_torch_xla_available():