manbeast3b commited on
Commit
ef90742
·
verified ·
1 Parent(s): 4bc5819

Update src/pipeline.py

Browse files
Files changed (1) hide show
  1. 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(