Update README.md
Browse files
README.md
CHANGED
|
@@ -116,6 +116,12 @@ response = tokenizer.decode(outputs[0][inputs['input_ids'].shape[1]:], skip_spec
|
|
| 116 |
print(response)
|
| 117 |
```
|
| 118 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 119 |
## Training Results
|
| 120 |
|
| 121 |
- **Initial Loss**: 2.95 (PPL: 19.08)
|
|
|
|
| 116 |
print(response)
|
| 117 |
```
|
| 118 |
|
| 119 |
+
Works with vllm.
|
| 120 |
+
|
| 121 |
+
```
|
| 122 |
+
vllm serve retowyss/PromptBridge-0.6b-Alpha
|
| 123 |
+
```
|
| 124 |
+
|
| 125 |
## Training Results
|
| 126 |
|
| 127 |
- **Initial Loss**: 2.95 (PPL: 19.08)
|