Fix pipeline tag, correct metadata typo, and add paper link
#1
by
nielsr
HF staff
- opened
README.md
CHANGED
@@ -1,18 +1,17 @@
|
|
1 |
---
|
2 |
base_model:
|
3 |
- Qwen/Qwen2-VL-2B-Instruct
|
|
|
|
|
|
|
|
|
4 |
library_name: transformers
|
|
|
5 |
model_name: HazardNet-unsloth-v0.4
|
|
|
6 |
tags:
|
7 |
- trl
|
8 |
- sft
|
9 |
-
licence: license
|
10 |
-
license: apache-2.0
|
11 |
-
datasets:
|
12 |
-
- Tami3/HazardQA
|
13 |
-
language:
|
14 |
-
- en
|
15 |
-
pipeline_tag: visual-question-answering
|
16 |
---
|
17 |
|
18 |
# Model Card for HazardNet-unsloth-v0.4
|
@@ -20,6 +19,8 @@ pipeline_tag: visual-question-answering
|
|
20 |
This model is a fine-tuned version of [Qwen/Qwen2-VL-2B-Instruct](https://huggingface.co/Qwen/Qwen2-VL-2B-Instruct).
|
21 |
It has been trained using [TRL](https://github.com/huggingface/trl).
|
22 |
|
|
|
|
|
23 |
## Quick start
|
24 |
|
25 |
```python
|
@@ -30,7 +31,7 @@ from io import BytesIO
|
|
30 |
|
31 |
# Initialize the Visual Question Answering pipeline with HazardNet
|
32 |
hazard_vqa = pipeline(
|
33 |
-
"
|
34 |
model="Tami3/HazardNet"
|
35 |
)
|
36 |
|
@@ -82,7 +83,6 @@ except Exception as e:
|
|
82 |
|
83 |
|
84 |
|
85 |
-
|
86 |
This model was trained with SFT.
|
87 |
|
88 |
### Framework versions
|
@@ -95,8 +95,6 @@ This model was trained with SFT.
|
|
95 |
|
96 |
## Citations
|
97 |
|
98 |
-
|
99 |
-
|
100 |
Cite TRL as:
|
101 |
|
102 |
```bibtex
|
|
|
1 |
---
|
2 |
base_model:
|
3 |
- Qwen/Qwen2-VL-2B-Instruct
|
4 |
+
datasets:
|
5 |
+
- Tami3/HazardQA
|
6 |
+
language:
|
7 |
+
- en
|
8 |
library_name: transformers
|
9 |
+
license: apache-2.0
|
10 |
model_name: HazardNet-unsloth-v0.4
|
11 |
+
pipeline_tag: image-text-to-text
|
12 |
tags:
|
13 |
- trl
|
14 |
- sft
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
15 |
---
|
16 |
|
17 |
# Model Card for HazardNet-unsloth-v0.4
|
|
|
19 |
This model is a fine-tuned version of [Qwen/Qwen2-VL-2B-Instruct](https://huggingface.co/Qwen/Qwen2-VL-2B-Instruct).
|
20 |
It has been trained using [TRL](https://github.com/huggingface/trl).
|
21 |
|
22 |
+
The model was presented in the paper [](https://hf.co/papers/2502.20572).
|
23 |
+
|
24 |
## Quick start
|
25 |
|
26 |
```python
|
|
|
31 |
|
32 |
# Initialize the Visual Question Answering pipeline with HazardNet
|
33 |
hazard_vqa = pipeline(
|
34 |
+
"image-text-to-text",
|
35 |
model="Tami3/HazardNet"
|
36 |
)
|
37 |
|
|
|
83 |
|
84 |
|
85 |
|
|
|
86 |
This model was trained with SFT.
|
87 |
|
88 |
### Framework versions
|
|
|
95 |
|
96 |
## Citations
|
97 |
|
|
|
|
|
98 |
Cite TRL as:
|
99 |
|
100 |
```bibtex
|