Update src/pipeline.py
Browse files- src/pipeline.py +0 -7
src/pipeline.py
CHANGED
|
@@ -34,13 +34,6 @@ def load_pipeline() -> Pipeline:
|
|
| 34 |
"city96/t5-v1_1-xxl-encoder-bf16", torch_dtype=DTYPE
|
| 35 |
)
|
| 36 |
|
| 37 |
-
pipeline = DiffusionPipeline.from_pretrained(
|
| 38 |
-
MODEL_ID,
|
| 39 |
-
text_encoder=text_encoder,
|
| 40 |
-
text_encoder_2=text_encoder_2,
|
| 41 |
-
vae=vae,
|
| 42 |
-
torch_dtype=dtype,
|
| 43 |
-
)
|
| 44 |
pipeline = FluxPipeline.from_pretrained(MODEL_ID,vae=vae,text_encoder=text_encoder,text_encoder_2=text_encoder_2,
|
| 45 |
torch_dtype=DTYPE)
|
| 46 |
pipeline.text_encoder.to(memory_format=torch.channels_last)
|
|
|
|
| 34 |
"city96/t5-v1_1-xxl-encoder-bf16", torch_dtype=DTYPE
|
| 35 |
)
|
| 36 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 37 |
pipeline = FluxPipeline.from_pretrained(MODEL_ID,vae=vae,text_encoder=text_encoder,text_encoder_2=text_encoder_2,
|
| 38 |
torch_dtype=DTYPE)
|
| 39 |
pipeline.text_encoder.to(memory_format=torch.channels_last)
|