jlens-looped-transformers / code /prep_prompts.py
zigzag9939's picture
Add files using upload-large-folder tool
e418394 verified
Raw History Blame Contribute Delete
217 Bytes
import json
from jlens.examples import load_wikitext_prompts
prompts = load_wikitext_prompts(n_prompts=1000)
with open("prompts_1000.json", "w") as f:
json.dump(prompts, f)
print(f"saved {len(prompts)} prompts")