GregorZiegltrumAA
commited on
Commit
•
e13cf2c
1
Parent(s):
aef2844
Update README.md
Browse files
README.md
CHANGED
@@ -48,7 +48,7 @@ ckpt_path = Path("<path_to_repo>/umup-research-3b-fp8")
|
|
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 |
|