mskrt commited on
Commit
f94abeb
verified
1 Parent(s): e4e13db

Update pipeline.py

Browse files
Files changed (1) hide show
  1. pipeline.py +2 -1
pipeline.py CHANGED
@@ -51,7 +51,8 @@ class SuperDiffPipeline(DiffusionPipeline, ConfigMixin):
51
  scheduler=scheduler,
52
  vae=vae,
53
  text_encoder=text_encoder,
54
- tokenizer=tokenizer,)
 
55
 
56
  #self.register_to_config(
57
  # vae=vae.__class__.__name__,
 
51
  scheduler=scheduler,
52
  vae=vae,
53
  text_encoder=text_encoder,
54
+ tokenizer=tokenizer,
55
+ kwargs=kwargs)
56
 
57
  #self.register_to_config(
58
  # vae=vae.__class__.__name__,