patrickvonplaten commited on
Commit
b8fa126
1 Parent(s): f1a59e2

Update pipeline.py

Browse files
Files changed (1) hide show
  1. pipeline.py +1 -1
pipeline.py CHANGED
@@ -99,4 +99,4 @@ class CustomPipeline(DiffusionPipeline):
99
  if not return_dict:
100
  return (image,)
101
 
102
- return ImagePipelineOutput(images=image)
 
99
  if not return_dict:
100
  return (image,)
101
 
102
+ return ImagePipelineOutput(images=image), "This is a test"