Update README.md
Browse files
README.md
CHANGED
@@ -26,7 +26,7 @@ import torch
|
|
26 |
from transformers import PaliGemmaProcessor, PaliGemmaForConditionalGeneration, BitsAndBytesConfig, TrainingArguments, Trainer
|
27 |
import time
|
28 |
|
29 |
-
model = PaliGemmaForConditionalGeneration.from_pretrained('NYUAD-ComNets/
|
30 |
|
31 |
input_text ="extract the text from the image"
|
32 |
|
|
|
26 |
from transformers import PaliGemmaProcessor, PaliGemmaForConditionalGeneration, BitsAndBytesConfig, TrainingArguments, Trainer
|
27 |
import time
|
28 |
|
29 |
+
model = PaliGemmaForConditionalGeneration.from_pretrained('NYUAD-ComNets/VehiclePaliGemma',torch_dtype=torch.bfloat16)
|
30 |
|
31 |
input_text ="extract the text from the image"
|
32 |
|