Refine readme

Files changed (4) hide show

.gitattributes CHANGED Viewed

@@ -33,4 +33,4 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
 *.zip filter=lfs diff=lfs merge=lfs -text
 *.zst filter=lfs diff=lfs merge=lfs -text
 *tfevents* filter=lfs diff=lfs merge=lfs -text
-comparison.png filter=lfs diff=lfs merge=lfs -text

 *.zip filter=lfs diff=lfs merge=lfs -text
 *.zst filter=lfs diff=lfs merge=lfs -text
 *tfevents* filter=lfs diff=lfs merge=lfs -text
+*.png filter=lfs diff=lfs merge=lfs -text

README.md CHANGED Viewed

@@ -55,7 +55,7 @@ text = f"A chat between a curious user and an artificial intelligence assistant.
 text_chunks = [tokenizer(chunk).input_ids for chunk in text.split('<image>')]
 input_ids = torch.tensor(text_chunks[0] + [-200] + text_chunks[1], dtype=torch.long).unsqueeze(0)
-# image
 image = Image.open('example_2.png')
 image_tensor = model.process_images([image], model.config).to(dtype=model.dtype)

 text_chunks = [tokenizer(chunk).input_ids for chunk in text.split('<image>')]
 input_ids = torch.tensor(text_chunks[0] + [-200] + text_chunks[1], dtype=torch.long).unsqueeze(0)
+# image, sample images can be found in https://huggingface.co/BAAI/bunny-phi-2-siglip/tree/main/images
 image = Image.open('example_2.png')
 image_tensor = model.process_images([image], model.config).to(dtype=model.dtype)

images/example_1.png ADDED Viewed

images/example_2.png ADDED Viewed