Update src/pipeline.py
Browse files- src/pipeline.py +4 -0
src/pipeline.py
CHANGED
@@ -590,6 +590,10 @@ def empty_cache():
|
|
590 |
def load_pipeline() -> Pipeline:
|
591 |
empty_cache()
|
592 |
|
|
|
|
|
|
|
|
|
593 |
dtype, device = torch.bfloat16, "cuda"
|
594 |
|
595 |
text_encoder_2 = T5EncoderModel.from_pretrained(
|
|
|
590 |
def load_pipeline() -> Pipeline:
|
591 |
empty_cache()
|
592 |
|
593 |
+
from diffusers import FluxPipeline, FluxTransformer2DModel, GGUFQuantizationConfig
|
594 |
+
from diffusers.loaders.single_file_utils import create_diffusers_t5_model_from_checkpoint
|
595 |
+
from diffusers.loaders.single_file_model import FromOriginalModelMixin
|
596 |
+
|
597 |
dtype, device = torch.bfloat16, "cuda"
|
598 |
|
599 |
text_encoder_2 = T5EncoderModel.from_pretrained(
|