Update pipeline.py
Browse files- pipeline.py +1 -1
pipeline.py
CHANGED
|
@@ -1618,4 +1618,4 @@ class LGMFullPipeline(DiffusionPipeline):
|
|
| 1618 |
images = images.half().cuda()
|
| 1619 |
|
| 1620 |
result = self.lgm(images)
|
| 1621 |
-
return result
|
|
|
|
| 1618 |
images = images.half().cuda()
|
| 1619 |
|
| 1620 |
result = self.lgm(images)
|
| 1621 |
+
return result,images
|