Update src/pipeline.py
Browse files- src/pipeline.py +1 -1
src/pipeline.py
CHANGED
|
@@ -118,7 +118,7 @@ def load_pipeline():
|
|
| 118 |
)
|
| 119 |
|
| 120 |
# pipeline
|
| 121 |
-
pipeline = DiffusionPipeline(
|
| 122 |
CKPT_ID,
|
| 123 |
transformer=transformer_model,
|
| 124 |
text_encoder=text_encoder,
|
|
|
|
| 118 |
)
|
| 119 |
|
| 120 |
# pipeline
|
| 121 |
+
pipeline = DiffusionPipeline.from_pretrained(
|
| 122 |
CKPT_ID,
|
| 123 |
transformer=transformer_model,
|
| 124 |
text_encoder=text_encoder,
|