Update seed2_tokenizer.py

seed2_tokenizer.py

@@ -2136,18 +2136,13 @@ class Seed2Tokenizer(PreTrainedModel):
             id, _ = self.model.get_codebook_indices(visual_encoder, img)
         return id.view(img.shape[0], -1)
 
-    def decode(self, diffusion_model, indices, negative_indices=None, guidance_scale=10, num_inference_steps=20):
+    def decode(self, diffusion_model, indices,  guidance_scale=10, num_inference_steps=20):
         image_embeds = self.model.get_codebook_entry(indices)
         # image = self.diffusion_model(image_embeds=image_embed,
         #                              noise_level=0,
         #                              num_inference_steps=20,
         #                              latents=self.latents,
         #                              noise=self.noise).images
-        if negative_indices is not None:
-            assert indices.shape == negative_indices.shape, 'Negative indices must have the same shape with indices'
-            negative_image_embeds = self.model.get_codebook_entry(negative_indices)
-        else:
-            negative_image_embeds = None
 
         image = diffusion_model(
             image_embeds=image_embeds,