Update README.md

README.md

@@ -52,6 +52,10 @@ print("device_index:",device_index)
 base_model = PaliGemmaForConditionalGeneration.from_pretrained("BhashaAI/ViLaH",device_map={"": device_index},torch_dtype=torch.float16,low_cpu_mem_usage=True)
 processor = AutoProcessor.from_pretrained("BhashaAI/ViLaH")
 
+inputs = processor(text=text, images=test_image, return_tensors="pt").to("cuda")
+for k,v in inputs.items():
+  print(k,v.shape)
+
 MAX_LENGTH = 500
 # Autoregressively generate
 # We use greedy decoding here, for more fancy methods see https://huggingface.co/blog/how-to-generate
@@ -87,6 +91,10 @@ quantization_config = BitsAndBytesConfig(load_in_4bit=True)
 base_model = PaliGemmaForConditionalGeneration.from_pretrained("BhashaAI/ViLaH",device_map={"": device_index},quantization_config=quantization_config,torch_dtype=torch.float16,low_cpu_mem_usage=True)
 processor = AutoProcessor.from_pretrained("BhashaAI/ViLaH")
 
+inputs = processor(text=text, images=test_image, return_tensors="pt").to("cuda")
+for k,v in inputs.items():
+  print(k,v.shape)
+
 MAX_LENGTH = 500
 # Autoregressively generate
 # We use greedy decoding here, for more fancy methods see https://huggingface.co/blog/how-to-generate