Nits for model docs (#29795)

* Update llava_next.md

* Update seggpt.md
This commit is contained in:
Merve Noyan
2024-04-22 12:41:03 +03:00
committed by GitHub
parent 8c12690cec
commit b20b017949
2 changed files with 3 additions and 3 deletions

View File

@@ -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
- forward