manbeast3b commited on
Commit
f53ad1f
·
verified ·
1 Parent(s): 5560a1b

Update src/pipeline.py

Browse files
Files changed (1) hide show
  1. src/pipeline.py +2 -4
src/pipeline.py CHANGED
@@ -626,19 +626,17 @@ def empty_cache():
626
  def load_pipeline() -> Pipeline:
627
  empty_cache()
628
 
629
- from diffusers import FluxPipeline, FluxTransformer2DModel, GGUFQuantizationConfig
630
- from diffusers.loaders.single_file_utils import create_diffusers_t5_model_from_checkpoint
631
- from diffusers.loaders.single_file_model import FromOriginalModelMixin
632
 
633
 
634
  dtype, device = torch.bfloat16, "cuda"
635
 
636
  import pdb; pdb.set_trace()
637
-
638
  t5_path = os.path.join(HF_HUB_CACHE, "models--manbeast3b--t5-v1_1-xxl-encoder-q8/snapshots/59c6c9cb99dcea42067f32caac3ea0836ef4c548/t5-v1_1-xxl-encoder-Q8_0.gguf")
639
  # config_path = os.path.join(HF_HUB_CACHE, "models--black-forest--labs/FLUX.1-schnell/snapshots/741f7c3ce8b383c54771c7003378a50191e9efe9/text_encoder_2/config.json")
640
  config_path = os.path.join(HF_HUB_CACHE, "models--black-forest-labs--FLUX.1-schnell/snapshots/741f7c3ce8b383c54771c7003378a50191e9efe9/")
641
  ckpt_t5 = load_single_file_checkpoint(t5_path,local_files_only=True)
 
642
 
643
  text_encoder_2 = T5EncoderModel.from_pretrained(
644
  "silentdriver/aadb864af9", revision = "060dabc7fa271c26dfa3fd43c16e7c5bf3ac7892", torch_dtype=torch.bfloat16
 
626
  def load_pipeline() -> Pipeline:
627
  empty_cache()
628
 
629
+
 
 
630
 
631
 
632
  dtype, device = torch.bfloat16, "cuda"
633
 
634
  import pdb; pdb.set_trace()
 
635
  t5_path = os.path.join(HF_HUB_CACHE, "models--manbeast3b--t5-v1_1-xxl-encoder-q8/snapshots/59c6c9cb99dcea42067f32caac3ea0836ef4c548/t5-v1_1-xxl-encoder-Q8_0.gguf")
636
  # config_path = os.path.join(HF_HUB_CACHE, "models--black-forest--labs/FLUX.1-schnell/snapshots/741f7c3ce8b383c54771c7003378a50191e9efe9/text_encoder_2/config.json")
637
  config_path = os.path.join(HF_HUB_CACHE, "models--black-forest-labs--FLUX.1-schnell/snapshots/741f7c3ce8b383c54771c7003378a50191e9efe9/")
638
  ckpt_t5 = load_single_file_checkpoint(t5_path,local_files_only=True)
639
+ print("the file is loaded")
640
 
641
  text_encoder_2 = T5EncoderModel.from_pretrained(
642
  "silentdriver/aadb864af9", revision = "060dabc7fa271c26dfa3fd43c16e7c5bf3ac7892", torch_dtype=torch.bfloat16