Upload 13 files
Browse files- pipeline.py +2 -3
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
|
14 |
"""__init__.
|
15 |
|
16 |
Parameters
|
@@ -51,8 +51,7 @@ class SuperDiffPipeline(DiffusionPipeline, ConfigMixin):
|
|
51 |
scheduler=scheduler,
|
52 |
vae=vae,
|
53 |
text_encoder=text_encoder,
|
54 |
-
tokenizer=tokenizer,
|
55 |
-
**kwargs)
|
56 |
|
57 |
|
58 |
#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) -> None:
|
14 |
"""__init__.
|
15 |
|
16 |
Parameters
|
|
|
51 |
scheduler=scheduler,
|
52 |
vae=vae,
|
53 |
text_encoder=text_encoder,
|
54 |
+
tokenizer=tokenizer,)
|
|
|
55 |
|
56 |
|
57 |
#self.register_to_config(
|