Improve Model Card: Correct pipeline tag and add library name (#1)
Browse files- Improve Model Card: Correct pipeline tag and add library name (e8f0f6ab7feb4788cec5daeeed7c44cadc48292b)
Co-authored-by: Niels Rogge <[email protected]>
README.md
CHANGED
@@ -1,13 +1,14 @@
|
|
1 |
---
|
2 |
-
license: apache-2.0
|
3 |
-
datasets:
|
4 |
-
- THUdyh/Oryx-SFT-Data
|
5 |
base_model:
|
6 |
- 01-ai/Yi-1.5-34B
|
7 |
-
|
|
|
8 |
language:
|
9 |
- en
|
10 |
- zh
|
|
|
|
|
|
|
11 |
---
|
12 |
|
13 |
# Oryx-34B
|
@@ -26,7 +27,7 @@ Oryx offers an on-demand solution to seamlessly and efficiently process visual i
|
|
26 |
|
27 |
We provide a simple generation process for using our model. For more details, please refer to our [Github Repo](https://github.com/liuzuyan/oryx)
|
28 |
|
29 |
-
```
|
30 |
from oryx.model.builder import load_pretrained_model
|
31 |
from oryx.mm_utils import get_model_name_from_path, process_images, tokenizer_image_token
|
32 |
from oryx.constants import IMAGE_TOKEN_INDEX, DEFAULT_IMAGE_TOKEN, DEFAULT_IM_START_TOKEN, DEFAULT_IM_END_TOKEN, IGNORE_INDEX
|
|
|
1 |
---
|
|
|
|
|
|
|
2 |
base_model:
|
3 |
- 01-ai/Yi-1.5-34B
|
4 |
+
datasets:
|
5 |
+
- THUdyh/Oryx-SFT-Data
|
6 |
language:
|
7 |
- en
|
8 |
- zh
|
9 |
+
license: apache-2.0
|
10 |
+
pipeline_tag: video-text-to-text
|
11 |
+
library_name: transformers
|
12 |
---
|
13 |
|
14 |
# Oryx-34B
|
|
|
27 |
|
28 |
We provide a simple generation process for using our model. For more details, please refer to our [Github Repo](https://github.com/liuzuyan/oryx)
|
29 |
|
30 |
+
```python
|
31 |
from oryx.model.builder import load_pretrained_model
|
32 |
from oryx.mm_utils import get_model_name_from_path, process_images, tokenizer_image_token
|
33 |
from oryx.constants import IMAGE_TOKEN_INDEX, DEFAULT_IMAGE_TOKEN, DEFAULT_IM_START_TOKEN, DEFAULT_IM_END_TOKEN, IGNORE_INDEX
|