mskrt commited on
Commit
c325aa7
verified
1 Parent(s): d06cb8a

Update pipeline.py

Browse files
Files changed (1) hide show
  1. pipeline.py +1 -1
pipeline.py CHANGED
@@ -50,7 +50,7 @@ class SuperDiffPipeline(DiffusionPipeline, ConfigMixin):
50
  scheduler=scheduler.__class__.__name__,
51
  tokenizer=tokenizer.__class__.__name__,
52
  unet=unet.__class__.__name__,
53
- text_encoder=text_encoder.__class__.__name__
54
  device=device,
55
  batch_size=None,
56
  num_inference_steps=None,
 
50
  scheduler=scheduler.__class__.__name__,
51
  tokenizer=tokenizer.__class__.__name__,
52
  unet=unet.__class__.__name__,
53
+ text_encoder=text_encoder.__class__.__name__,
54
  device=device,
55
  batch_size=None,
56
  num_inference_steps=None,