Update README.md

README.md

@@ -1,10 +1,18 @@
 ```# first git clone this repo
 import os
+import torch
+if not os.path.exists("seed2"):
+    os.system("git clone https://huggingface.co/ontocord/seed2")
 from seed2.seed2_tokenizer import Seed2Tokenizer
-tokenizer = Seed2Tokenizer.from_pretrained("seed2")
+tokenizer = Seed2Tokenizer.from_pretrained("seed2", torch_dtype=torch.float16).to('cuda')
 if not os.path.exists("cat.jpg"):
     os.system("wget https://images.unsplash.com/photo-1574158622682-e40e69881006?w=300 -O cat.jpg")
 tokens = tokenizer.encode_image("cat.jpg")
 print (tokens)
 image_embeds = tokenizer.model.get_codebook_entry(tokens)
-print (image_embeds)   ```
+print (image_embeds) 
+from diffusers import StableUnCLIPImg2ImgPipeline
+pipe = StableUnCLIPImg2ImgPipeline.from_pretrained('stabilityai/stable-diffusion-2-1-unclip', torch_dtype=torch.float16)
+pipe = pipe.to('cuda')
+tokenizer.decode(pipe, tokens)[0]
+```