Update README.md
#10
by
elischwartz
- opened
README.md
CHANGED
|
@@ -151,6 +151,12 @@ outputs = model.generate(inputs, sampling_params=sampling_params)
|
|
| 151 |
print(f"Generated text: {outputs[0].outputs[0].text}")
|
| 152 |
```
|
| 153 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 154 |
**Model Architecture:**
|
| 155 |
|
| 156 |
The architecture of granite-vision-3.1-2b-preview consists of the following components:
|
|
|
|
| 151 |
print(f"Generated text: {outputs[0].outputs[0].text}")
|
| 152 |
```
|
| 153 |
|
| 154 |
+
### Fine-tuning
|
| 155 |
+
|
| 156 |
+
For an example of fine-tuning Granite Vision for new tasks refer to [this notebook](https://huggingface.co/learn/cookbook/en/fine_tuning_granite_vision_sft_trl).
|
| 157 |
+
|
| 158 |
+
|
| 159 |
+
|
| 160 |
**Model Architecture:**
|
| 161 |
|
| 162 |
The architecture of granite-vision-3.1-2b-preview consists of the following components:
|