Update seed2_tokenizer.py

seed2_tokenizer.py

@@ -2448,7 +2448,7 @@ class Seed2Tokenizer(PreTrainedModel):
             id = self.model.get_codebook_indices_only(visual_encoder, img)
         return id.view(img.shape[0], -1)
 
-    def decode(self, diffusion_model, indices, guidance_scale=10, noise_level=0, num_inference_steps=10,):
+    def decode(self, diffusion_model, indices, guidance_scale=10, noise_level=0, num_inference_steps=20,):
         image_embeds = self.model.get_codebook_entry(indices)
         image_embeds = image_embeds.to(dtype=diffusion_model.dtype, device=diffusion_model.device)
         image = diffusion_model(