Update README.md
Browse files
README.md
CHANGED
@@ -27,9 +27,6 @@ pip install git+https://github.com/TIGER-AI-Lab/Mantis.git
|
|
27 |
```
|
28 |
then run inference with codes here: [examples/run_mantis.py](https://github.com/TIGER-AI-Lab/Mantis/blob/main/examples/run_mantis_hf.py)
|
29 |
|
30 |
-
Or, you can run the model without relying on the mantis codes, using pure hugging face transformers. See [examples/run_mantis_hf.py](https://github.com/TIGER-AI-Lab/Mantis/blob/main/examples/run_mantis_hf.py) for details.
|
31 |
-
|
32 |
-
|
33 |
```python
|
34 |
from mantis.models.mllava import chat_mllava
|
35 |
from PIL import Image
|
@@ -61,5 +58,8 @@ print("ASSISTANT: ", response)
|
|
61 |
# There are two items in image 1 and four items in image 2.
|
62 |
```
|
63 |
|
|
|
|
|
|
|
64 |
## Training
|
65 |
Training codes will be released soon.
|
|
|
27 |
```
|
28 |
then run inference with codes here: [examples/run_mantis.py](https://github.com/TIGER-AI-Lab/Mantis/blob/main/examples/run_mantis_hf.py)
|
29 |
|
|
|
|
|
|
|
30 |
```python
|
31 |
from mantis.models.mllava import chat_mllava
|
32 |
from PIL import Image
|
|
|
58 |
# There are two items in image 1 and four items in image 2.
|
59 |
```
|
60 |
|
61 |
+
Or, you can run the model without relying on the mantis codes, using pure hugging face transformers. See [examples/run_mantis_hf.py](https://github.com/TIGER-AI-Lab/Mantis/blob/main/examples/run_mantis_hf.py) for details.
|
62 |
+
|
63 |
+
|
64 |
## Training
|
65 |
Training codes will be released soon.
|