Update src/pipeline.py
Browse files- 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 |
-
|
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
|