retowyss commited on
Commit
834d2b3
·
verified ·
1 Parent(s): efb119d

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +6 -0
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)