mskrt commited on
Commit
cf1bda4
verified
1 Parent(s): ab0899e

Update pipeline.py

Browse files
Files changed (1) hide show
  1. 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(