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(
|