Update example_03_chaos_mode.py

This commit is contained in:
PSBigBig 2025-06-11 14:14:37 +08:00 committed by GitHub
parent b32bded342
commit bc40b0f4ec
No known key found for this signature in database
GPG key ID: B5690EEEBB952194

View file

@ -3,12 +3,20 @@ from wfgy_core import WFGYRunner
if __name__ == "__main__":
parser = argparse.ArgumentParser()
parser.add_argument("--reflect", action="store_true", help="Enable self-reflection")
parser.add_argument("--ascii", action="store_true", help="Show ASCII banner")
parser.add_argument("--style", type=str, default="meme", help="Choose style: default/scientific/meme")
parser.add_argument("--reflect", action="store_true", help="Enable self-reflection output")
parser.add_argument("--ascii", action="store_true", help="Enable ASCII visual output")
parser.add_argument("--style", type=str, default="chaos", help="Output style mode")
parser.add_argument("--model", type=str, default="gpt2-medium", help="Model ID (e.g., gpt2, gpt2-xl)")
args = parser.parse_args()
config = {
"style": args.style,
"reflect": args.reflect,
"ascii": args.ascii,
"model_id": args.model
}
runner = WFGYRunner(config=config)
prompt = "What happens if Schrödingers cat becomes a quantum influencer?"
runner = WFGYRunner()
runner.run(prompt, reflect=args.reflect, style=args.style, show_ascii=args.ascii)
output = runner.run(prompt)
print(output)