Fix generation from the published artifact; rewrite model card

modeling_baguettotron_vlm.py

@@ -67,16 +67,15 @@ class BaguettotronVLMForConditionalGeneration(PreTrainedModel):
     """
     BaguettotronVLM: InternViT-300M + PixelUnshuffle projector + Baguettotron-321M.
 
-    ~628M total parameters. The same architecture is shipped for all three
-    training stages; only the checkpoint weights and `config.chat_style`
-    differ between them.
+    ~628M total parameters. The same architecture is shipped for every
+    checkpoint; only the weights and `config.chat_style` differ.
 
     Load with:
         from transformers import AutoModelForImageTextToText
         model = AutoModelForImageTextToText.from_pretrained(
-            "andreagemelli/Baguettotron-VLM",
+            "andreagemelli/baguettotron-vision-vqa",
             trust_remote_code=True,
-            torch_dtype=torch.bfloat16,
+            dtype=torch.bfloat16,
         )
     """
 

processing_baguettotron_vlm.py

@@ -52,7 +52,7 @@ class BaguettotronVLMProcessor(ProcessorMixin):
 
         from transformers import AutoProcessor
         processor = AutoProcessor.from_pretrained(
-            "andreagemelli/Baguettotron-VLM",
+            "andreagemelli/baguettotron-vision-vqa",
             trust_remote_code=True,
         )
     """