Update pipeline.py
Browse files- pipeline.py +2 -2
pipeline.py
CHANGED
|
@@ -14,8 +14,8 @@ class ScoreSdeVePipeline(DiffusionPipeline):
|
|
| 14 |
A `ScoreSdeVeScheduler` to be used in combination with `unet` to denoise the encoded image.
|
| 15 |
"""
|
| 16 |
|
| 17 |
-
unet
|
| 18 |
-
scheduler
|
| 19 |
|
| 20 |
def __init__(self, unet, scheduler):
|
| 21 |
super().__init__()
|
|
|
|
| 14 |
A `ScoreSdeVeScheduler` to be used in combination with `unet` to denoise the encoded image.
|
| 15 |
"""
|
| 16 |
|
| 17 |
+
unet
|
| 18 |
+
scheduler
|
| 19 |
|
| 20 |
def __init__(self, unet, scheduler):
|
| 21 |
super().__init__()
|