mirror of
https://github.com/karpathy/llm.c.git
synced 2026-07-27 20:25:09 -04:00
Remove prompts.json
This commit is contained in:
parent
624ed3ce30
commit
dfd459bfc4
1 changed files with 11 additions and 1 deletions
|
|
@ -1192,7 +1192,17 @@ if __name__ == "__main__":
|
|||
and (step % args.sample_every == 0 or last_step)) \
|
||||
and master_process:
|
||||
model.eval()
|
||||
prompts: List[str] = json.loads(open(os.path.join(os.path.dirname(__file__), 'llmc_py', 'prompts.json')).read())['prompts']
|
||||
prompts: List[str] = [
|
||||
"Clearly, the meaning of life is",
|
||||
"Simply put, the theory of relativity states that",
|
||||
"""The repo llm.c on GitHub is""",
|
||||
"""Translate English to French:
|
||||
|
||||
sea otter => loutre de mer
|
||||
peppermint => menthe poivrée
|
||||
plush girafe => girafe peluche
|
||||
cheese =>""",
|
||||
]
|
||||
if args.use_hf:
|
||||
prompt_tokens = [model.tokenizer(x).input_ids for x in prompts]
|
||||
else: # Meta
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue