Update README.md
Browse files
README.md
CHANGED
|
@@ -48,7 +48,7 @@ ckpt_path = Path("<path_to_repo>/sp-baseline-research-7b-bf16")
|
|
| 48 |
|
| 49 |
model = TransformerInferenceModule.from_checkpoint(ckpt_path)
|
| 50 |
|
| 51 |
-
prompt = "Yesterday I dreamt of
|
| 52 |
|
| 53 |
output = model.generate(max_tokens=100, input_text=prompt)
|
| 54 |
|
|
|
|
| 48 |
|
| 49 |
model = TransformerInferenceModule.from_checkpoint(ckpt_path)
|
| 50 |
|
| 51 |
+
prompt = "Yesterday I dreamt of"
|
| 52 |
|
| 53 |
output = model.generate(max_tokens=100, input_text=prompt)
|
| 54 |
|