osanseviero
/

dalle-mini-fork

Model card Files Files and versions

Omar Sanseviero commited on Aug 17, 2021

Commit

cdc4c33

·

1 Parent(s): b336e66

Fix

Files changed (1) hide show

pipeline.py +1 -1

pipeline.py CHANGED Viewed

@@ -93,7 +93,7 @@ class PreTrainedPipeline():
         # remove first token (BOS)
         encoded_image = encoded_image.sequences[..., 1:]
-        decoded_image = vqgan.decode_code(encoded_image)
         clipped_image = decoded_image.squeeze().clip(0., 1.)
         return Image.fromarray(np.asarray(clipped_image * 255, dtype=np.uint8))

         # remove first token (BOS)
         encoded_image = encoded_image.sequences[..., 1:]
+        decoded_image = self.vqgan.decode_code(encoded_image)
         clipped_image = decoded_image.squeeze().clip(0., 1.)
         return Image.fromarray(np.asarray(clipped_image * 255, dtype=np.uint8))