GregorZiegltrumAA
commited on
Commit
•
7c75041
1
Parent(s):
58b7532
Update README.md
Browse files
README.md
CHANGED
@@ -48,7 +48,7 @@ ckpt_path = Path("<path_to_repo>/sp-baseline-research-1b-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 |
|