Commit
•
cdc4c33
1
Parent(s):
b336e66
Fix
Browse files- pipeline.py +1 -1
pipeline.py
CHANGED
@@ -93,7 +93,7 @@ class PreTrainedPipeline():
|
|
93 |
|
94 |
# remove first token (BOS)
|
95 |
encoded_image = encoded_image.sequences[..., 1:]
|
96 |
-
decoded_image = vqgan.decode_code(encoded_image)
|
97 |
clipped_image = decoded_image.squeeze().clip(0., 1.)
|
98 |
|
99 |
return Image.fromarray(np.asarray(clipped_image * 255, dtype=np.uint8))
|
|
|
93 |
|
94 |
# remove first token (BOS)
|
95 |
encoded_image = encoded_image.sequences[..., 1:]
|
96 |
+
decoded_image = self.vqgan.decode_code(encoded_image)
|
97 |
clipped_image = decoded_image.squeeze().clip(0., 1.)
|
98 |
|
99 |
return Image.fromarray(np.asarray(clipped_image * 255, dtype=np.uint8))
|