Update README.md

README.md

@@ -41,7 +41,7 @@ def image_to_indices(image, vq):
   
     return indices, norm
 
-def norm_to_tensor(indices, vq, ada: AdaLayerNorm):
+def indices_to_tensor(indices, vq, ada: AdaLayerNorm):
     b = indices.shape[0]
     w = int(indices.shape[-1] ** 0.5)
     v = vq.quantize.embedding(indices)
@@ -61,7 +61,7 @@ if __name__ == '__main__':
     # Encode image.
     ids, _ = image_to_indices(image, vq)
     # Decode by integer ids.
-    pixel_values = tokenizer.norm_to_tensor(ids, vq, ada) / 2.0 + 0.5
+    pixel_values = indices_to_tensor(ids, vq, ada) / 2.0 + 0.5
     preview = transforms.functional.to_pil_image(pixel_values.detach().squeeze(0).to('cpu'))
 ```