mirror of
https://github.com/onestardao/WFGY.git
synced 2026-04-28 03:29:51 +00:00
Update example_03_chaos_mode.py
This commit is contained in:
parent
b32bded342
commit
bc40b0f4ec
1 changed files with 14 additions and 6 deletions
|
|
@ -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ödinger’s 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)
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue