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,
|