Update pipeline.py
Browse files- pipeline.py +2 -1
pipeline.py
CHANGED
|
@@ -10,7 +10,7 @@ from diffusers.configuration_utils import ConfigMixin
|
|
| 10 |
class SuperDiffPipeline(DiffusionPipeline, ConfigMixin):
|
| 11 |
"""SuperDiffPipeline."""
|
| 12 |
|
| 13 |
-
def __init__(self, unet: Callable, vae: Callable, text_encoder: Callable, scheduler: Callable, tokenizer: Callable) -> None:
|
| 14 |
"""__init__.
|
| 15 |
|
| 16 |
Parameters
|
|
@@ -51,6 +51,7 @@ class SuperDiffPipeline(DiffusionPipeline, ConfigMixin):
|
|
| 51 |
vae=vae,
|
| 52 |
text_encoder=text_encoder,
|
| 53 |
tokenizer=tokenizer,
|
|
|
|
| 54 |
)
|
| 55 |
|
| 56 |
#self.register_to_config(
|
|
|
|
| 10 |
class SuperDiffPipeline(DiffusionPipeline, ConfigMixin):
|
| 11 |
"""SuperDiffPipeline."""
|
| 12 |
|
| 13 |
+
def __init__(self, unet: Callable, vae: Callable, text_encoder: Callable, scheduler: Callable, tokenizer: Callable, **kwargs) -> None:
|
| 14 |
"""__init__.
|
| 15 |
|
| 16 |
Parameters
|
|
|
|
| 51 |
vae=vae,
|
| 52 |
text_encoder=text_encoder,
|
| 53 |
tokenizer=tokenizer,
|
| 54 |
+
**kwargs
|
| 55 |
)
|
| 56 |
|
| 57 |
#self.register_to_config(
|