Update pipeline.py
Browse files- pipeline.py +0 -1
pipeline.py
CHANGED
|
@@ -34,7 +34,6 @@ class PreTrainedPipeline():
|
|
| 34 |
|
| 35 |
# Ugly hack, I'm sure there's a better way of doing what I want.
|
| 36 |
img = output[0]
|
| 37 |
-
img = img.transpose((0, 2, 3, 1))
|
| 38 |
img = (img + 1) / 2.0)
|
| 39 |
img = transforms.ToPILImage()(img)
|
| 40 |
return img
|
|
|
|
| 34 |
|
| 35 |
# Ugly hack, I'm sure there's a better way of doing what I want.
|
| 36 |
img = output[0]
|
|
|
|
| 37 |
img = (img + 1) / 2.0)
|
| 38 |
img = transforms.ToPILImage()(img)
|
| 39 |
return img
|