NYUAD-ComNets
commited on
Update README.md
Browse files
README.md
CHANGED
@@ -23,7 +23,7 @@ import torch
|
|
23 |
from transformers import PaliGemmaProcessor, PaliGemmaForConditionalGeneration, BitsAndBytesConfig, TrainingArguments, Trainer
|
24 |
|
25 |
|
26 |
-
model = PaliGemmaForConditionalGeneration.from_pretrained('NYUAD-ComNets/
|
27 |
|
28 |
input_text = "what is age group of person in the image?"
|
29 |
|
|
|
23 |
from transformers import PaliGemmaProcessor, PaliGemmaForConditionalGeneration, BitsAndBytesConfig, TrainingArguments, Trainer
|
24 |
|
25 |
|
26 |
+
model = PaliGemmaForConditionalGeneration.from_pretrained('NYUAD-ComNets/FaceScanPaliGemma_Age',torch_dtype=torch.bfloat16)
|
27 |
|
28 |
input_text = "what is age group of person in the image?"
|
29 |
|