Unverified Commit b20b0179 authored by Merve Noyan's avatar Merve Noyan Committed by GitHub
Browse files

Nits for model docs (#29795)

* Update llava_next.md

* Update seggpt.md
parent 8c12690c
...@@ -98,7 +98,7 @@ print(processor.decode(output[0], skip_special_tokens=True)) ...@@ -98,7 +98,7 @@ print(processor.decode(output[0], skip_special_tokens=True))
### Quantization using Bitsandbytes ### 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 ```python
from transformers import LlavaNextForConditionalGeneration, BitsAndBytesConfig from transformers import LlavaNextForConditionalGeneration, BitsAndBytesConfig
......
...@@ -36,7 +36,7 @@ import torch ...@@ -36,7 +36,7 @@ import torch
from datasets import load_dataset from datasets import load_dataset
from transformers import SegGptImageProcessor, SegGptForImageSegmentation from transformers import SegGptImageProcessor, SegGptForImageSegmentation
model_id = "BAAI/seggpt-vit-large" checkpoint = "BAAI/seggpt-vit-large"
image_processor = SegGptImageProcessor.from_pretrained(checkpoint) image_processor = SegGptImageProcessor.from_pretrained(checkpoint)
model = SegGptForImageSegmentation.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 ...@@ -87,4 +87,4 @@ The original code can be found [here]([(https://github.com/baaivision/Painter/tr
## SegGptForImageSegmentation ## SegGptForImageSegmentation
[[autodoc]] SegGptForImageSegmentation [[autodoc]] SegGptForImageSegmentation
- forward - forward
\ No newline at end of file
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment