mskrt commited on
Commit
a8cf307
verified
1 Parent(s): 6a798c0

Upload 13 files

Browse files
Files changed (1) hide show
  1. 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, **kwargs) -> None:
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(