Update seed2_tokenizer.py

seed2_tokenizer.py

@@ -42,7 +42,7 @@ DIFFUSION_NAME = 'stabilityai/stable-diffusion-2-1-unclip'
 # from qformer.qformer_quantizer import Blip2QformerQuantizer
 # from diffusers import StableUnCLIPImg2ImgPipeline
 
-from pipeline_stable_unclip_img2img import StableUnCLIPImg2ImgPipeline
+from diffusers import StableUnCLIPImg2ImgPipeline
 
 import logging
 
@@ -2182,7 +2182,7 @@ class Seed2Tokenizer(PreTrainedModel):
             image_torch = self.processor(image_pil)
 
             image_torch = image_torch.to(self.device)
-        return self.encode(visual_encoder, image_torch)
+        return self.encode(image_torch, visual_encoder)
     
 if __name__ == "__main__":
     tokenizer = Seed2Tokenizer.from_pretrained("ontocord/seed2")