Skip to content

Commit

Permalink
Nits for model docs (#29795)
Browse files Browse the repository at this point in the history
* Update llava_next.md

* Update seggpt.md
  • Loading branch information
merveenoyan authored Apr 22, 2024
1 parent 8c12690 commit b20b017
Show file tree
Hide file tree
Showing 2 changed files with 3 additions and 3 deletions.
2 changes: 1 addition & 1 deletion docs/source/en/model_doc/llava_next.md
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand Down
4 changes: 2 additions & 2 deletions docs/source/en/model_doc/seggpt.md
Original file line number Diff line number Diff line change
Expand Up @@ -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)

Expand Down Expand Up @@ -87,4 +87,4 @@ The original code can be found [here]([(https://github.com/baaivision/Painter/tr
## SegGptForImageSegmentation

[[autodoc]] SegGptForImageSegmentation
- forward
- forward

0 comments on commit b20b017

Please sign in to comment.