manbeast3b
/

optimization1

Model card Files Files and versions

manbeast3b commited on Dec 10, 2024

Commit

ab415dd

·

verified ·

1 Parent(s): daa1640

Update src/pipeline.py

Files changed (1) hide show

src/pipeline.py +4 -3

src/pipeline.py CHANGED Viewed

@@ -42,9 +42,10 @@ def load_pipeline() -> Pipeline:
     # text_encoder_2 = T5EncoderModel.from_pretrained(
     #     "city96/t5-v1_1-xxl-encoder-bf16", torch_dtype=torch.bfloat16
     # )
-    text_encoder_2 = T5EncoderModel.from_pretrained(
-        "sayakpaul/flux.1-dev-nf4-pkg", subfolder="text_encoder_2", torch_dtype=torch.bfloat16
-    )
     vae=AutoencoderKL.from_pretrained(ckpt_id, subfolder="vae", torch_dtype=dtype)
     pipeline = DiffusionPipeline.from_pretrained(
         ckpt_id,

     # text_encoder_2 = T5EncoderModel.from_pretrained(
     #     "city96/t5-v1_1-xxl-encoder-bf16", torch_dtype=torch.bfloat16
     # )
+    # text_encoder_2 = T5EncoderModel.from_pretrained(
+    #     "sayakpaul/flux.1-dev-nf4-pkg", subfolder="text_encoder_2", torch_dtype=torch.bfloat16
+    # )
+    text_encoder_2 = T5EncoderModel.from_pretrained("hf-internal-testing/tiny-random-t5")
     vae=AutoencoderKL.from_pretrained(ckpt_id, subfolder="vae", torch_dtype=dtype)
     pipeline = DiffusionPipeline.from_pretrained(
         ckpt_id,