Update README.md

README.md

@@ -15,6 +15,17 @@ from transformers import AutoTokenizer, AutoProcessor
 from PIL import Image
 import requests
 
+model = MM_LLMs.from_pretrained(
+    'mesolitica/malaysian-tinyllama-1.1b-siglip-base-384-vision',
+    flash_attention = True,
+    dtype = torch.bfloat16,
+    torch_dtype = torch.bfloat16
+)
+_ = model.cuda()
+
+image_processor = AutoProcessor.from_pretrained('google/siglip-base-patch16-384')
+tokenizer = AutoTokenizer.from_pretrained('mesolitica/malaysian-tinyllama-1.1b-siglip-base-384-vision')
+
 def prepare_dataset(messages, images: List[str] = None):
     if images is not None:
         images = [Image.open(f).convert('RGB') for f in images]
@@ -34,17 +45,6 @@ def prepare_dataset(messages, images: List[str] = None):
     outputs['image_starts'] = torch.tensor([tokenizer.convert_tokens_to_ids('<image>')] * len(outputs['images']))
     return outputs
 
-model = MM_LLMs.from_pretrained(
-    'mesolitica/malaysian-tinyllama-1.1b-siglip-base-384-vision',
-    flash_attention = True,
-    dtype = torch.bfloat16,
-    torch_dtype = torch.bfloat16
-)
-_ = model.cuda()
-
-image_processor = AutoProcessor.from_pretrained('google/siglip-base-patch16-384')
-tokenizer = AutoTokenizer.from_pretrained('mesolitica/malaysian-tinyllama-1.1b-siglip-base-384-vision')
-
 with open('Persian-cat-breed.jpg', 'wb') as fopen:
     fopen.write(requests.get('https://cdn.beautifulnara.net/wp-content/uploads/2017/12/10201620/Persian-cat-breed.jpg').content)