From b20b01794923e0c78229b07645fe5de33e69b428 Mon Sep 17 00:00:00 2001 From: Merve Noyan Date: Mon, 22 Apr 2024 12:41:03 +0300 Subject: [PATCH] Nits for model docs (#29795) * Update llava_next.md * Update seggpt.md --- docs/source/en/model_doc/llava_next.md | 2 +- docs/source/en/model_doc/seggpt.md | 4 ++-- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/docs/source/en/model_doc/llava_next.md b/docs/source/en/model_doc/llava_next.md index ef74bf7e10..a2a3913fca 100644 --- a/docs/source/en/model_doc/llava_next.md +++ b/docs/source/en/model_doc/llava_next.md @@ -98,7 +98,7 @@ print(processor.decode(output[0], skip_special_tokens=True)) ### Quantization using Bitsandbytes -The model can be loaded in 8 or 4 bits, greatly reducing the memory requirements while maintaining the performance of the original model. First make sure to install bitsandbytes, `pip install bitsandbytes`` and make sure to have access to a CUDA compatible GPU device. Simply change the snippet above with: +The model can be loaded in 8 or 4 bits, greatly reducing the memory requirements while maintaining the performance of the original model. First make sure to install bitsandbytes, `pip install bitsandbytes` and make sure to have access to a CUDA compatible GPU device. Simply change the snippet above with: ```python from transformers import LlavaNextForConditionalGeneration, BitsAndBytesConfig diff --git a/docs/source/en/model_doc/seggpt.md b/docs/source/en/model_doc/seggpt.md index 707be24017..f821fc14a0 100644 --- a/docs/source/en/model_doc/seggpt.md +++ b/docs/source/en/model_doc/seggpt.md @@ -36,7 +36,7 @@ import torch from datasets import load_dataset from transformers import SegGptImageProcessor, SegGptForImageSegmentation -model_id = "BAAI/seggpt-vit-large" +checkpoint = "BAAI/seggpt-vit-large" image_processor = SegGptImageProcessor.from_pretrained(checkpoint) model = SegGptForImageSegmentation.from_pretrained(checkpoint) @@ -87,4 +87,4 @@ The original code can be found [here]([(https://github.com/baaivision/Painter/tr ## SegGptForImageSegmentation [[autodoc]] SegGptForImageSegmentation - - forward \ No newline at end of file + - forward