Update README.md
Browse files
README.md
CHANGED
@@ -532,6 +532,9 @@ The model was trained on [LLaVA-OneVision Dataset](https://huggingface.co/datase
|
|
532 |
**Feel free to share your generations in the Community tab!**
|
533 |
|
534 |
### Generation
|
|
|
|
|
|
|
535 |
```python
|
536 |
# pip install git+https://github.com/LLaVA-VL/LLaVA-NeXT.git
|
537 |
from llava.model.builder import load_pretrained_model
|
@@ -548,7 +551,7 @@ import sys
|
|
548 |
import warnings
|
549 |
|
550 |
warnings.filterwarnings("ignore")
|
551 |
-
pretrained = "lmms-lab/llava-onevision-qwen2-0.5b-
|
552 |
model_name = "llava_qwen"
|
553 |
device = "cuda"
|
554 |
device_map = "auto"
|
|
|
532 |
**Feel free to share your generations in the Community tab!**
|
533 |
|
534 |
### Generation
|
535 |
+
|
536 |
+
We provide the simple generation process for using our model. For more details, you could refer to [Github](https://github.com/LLaVA-VL/LLaVA-NeXT).
|
537 |
+
|
538 |
```python
|
539 |
# pip install git+https://github.com/LLaVA-VL/LLaVA-NeXT.git
|
540 |
from llava.model.builder import load_pretrained_model
|
|
|
551 |
import warnings
|
552 |
|
553 |
warnings.filterwarnings("ignore")
|
554 |
+
pretrained = "lmms-lab/llava-onevision-qwen2-0.5b-ov"
|
555 |
model_name = "llava_qwen"
|
556 |
device = "cuda"
|
557 |
device_map = "auto"
|