Update pipeline.py
Browse files- pipeline.py +1 -1
pipeline.py
CHANGED
|
@@ -50,7 +50,7 @@ class SuperDiffPipeline(DiffusionPipeline, ConfigMixin):
|
|
| 50 |
scheduler=scheduler.__class__.__name__,
|
| 51 |
tokenizer=tokenizer.__class__.__name__,
|
| 52 |
unet=unet.__class__.__name__,
|
| 53 |
-
text_encoder=text_encoder.__class__.__name__
|
| 54 |
device=device,
|
| 55 |
batch_size=None,
|
| 56 |
num_inference_steps=None,
|
|
|
|
| 50 |
scheduler=scheduler.__class__.__name__,
|
| 51 |
tokenizer=tokenizer.__class__.__name__,
|
| 52 |
unet=unet.__class__.__name__,
|
| 53 |
+
text_encoder=text_encoder.__class__.__name__,
|
| 54 |
device=device,
|
| 55 |
batch_size=None,
|
| 56 |
num_inference_steps=None,
|