Update seed2_tokenizer.py

seed2_tokenizer.py

@@ -2136,7 +2136,7 @@ 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,  guidance_scale=10, num_inference_steps=20):
+    def decode(self, diffusion_model, indices, guidance_scale=10, noise_level=0, num_inference_steps=10,):
         image_embeds = self.model.get_codebook_entry(indices)
         # image = self.diffusion_model(image_embeds=image_embed,
         #                              noise_level=0,
@@ -2148,7 +2148,7 @@ class Seed2Tokenizer(PreTrainedModel):
             image_embeds=image_embeds,
             negative_image_embeds=negative_image_embeds,
             guidance_scale=guidance_scale,
-            noise_level=0,
+            noise_level=noise_level,
             num_inference_steps=num_inference_steps,
             latents=self.latents,
         ).images