Update src/pipeline.py
Browse files- src/pipeline.py +4 -1
src/pipeline.py
CHANGED
|
@@ -30,8 +30,11 @@ def load_pipeline() -> Pipeline:
|
|
| 30 |
empty_cache()
|
| 31 |
dtype, device = torch.bfloat16, "cuda"
|
| 32 |
|
|
|
|
|
|
|
|
|
|
| 33 |
text_encoder_2 = T5EncoderModel.from_pretrained(
|
| 34 |
-
"
|
| 35 |
)
|
| 36 |
vae=AutoencoderKL.from_pretrained(ckpt_id, subfolder="vae", torch_dtype=dtype)
|
| 37 |
pipeline = DiffusionPipeline.from_pretrained(
|
|
|
|
| 30 |
empty_cache()
|
| 31 |
dtype, device = torch.bfloat16, "cuda"
|
| 32 |
|
| 33 |
+
# text_encoder_2 = T5EncoderModel.from_pretrained(
|
| 34 |
+
# "city96/t5-v1_1-xxl-encoder-bf16", torch_dtype=torch.bfloat16
|
| 35 |
+
# )
|
| 36 |
text_encoder_2 = T5EncoderModel.from_pretrained(
|
| 37 |
+
"sayakpaul/flux.1-dev-nf4-pkg", subfolder="text_encoder_2", torch_dtype=torch.bfloat16
|
| 38 |
)
|
| 39 |
vae=AutoencoderKL.from_pretrained(ckpt_id, subfolder="vae", torch_dtype=dtype)
|
| 40 |
pipeline = DiffusionPipeline.from_pretrained(
|