Text Generation
scaling
GregorZiegltrumAA commited on
Commit
7c75041
1 Parent(s): 58b7532

Update README.md

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