hf-internal-testing
/

diffusers-dummy-pipeline

Model card Files Files and versions Community

patrickvonplaten commited on Oct 5, 2022

Commit

f1a59e2

•

1 Parent(s): c74cf55

Update pipeline.py

Files changed (1) hide show

pipeline.py +1 -1

pipeline.py CHANGED Viewed

@@ -21,7 +21,7 @@ import torch
 from diffusers.pipeline_utils import DiffusionPipeline, ImagePipelineOutput
-class DDIMPipeline(DiffusionPipeline):
     r"""
     This model inherits from [`DiffusionPipeline`]. Check the superclass documentation for the generic methods the
     library implements for all the pipelines (such as downloading or saving, running on a particular device, etc.)

 from diffusers.pipeline_utils import DiffusionPipeline, ImagePipelineOutput
+class CustomPipeline(DiffusionPipeline):
     r"""
     This model inherits from [`DiffusionPipeline`]. Check the superclass documentation for the generic methods the
     library implements for all the pipelines (such as downloading or saving, running on a particular device, etc.)