vrr/WFGY
2
0
Fork 0
mirror of https://github.com/onestardao/WFGY.git synced 2026-04-29 20:20:01 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 4

Update example_01_basic_run.py

Browse source
This commit is contained in:
PSBigBig 2025-06-11 20:44:25 +08:00 • committed by GitHub
parent ecad664a93
commit 465a3c7ea7
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 8 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

12
examples/example_01_basic_run.py
View file

@ -1,15 +1,19 @@
# examples/example_01_basic_run.py
# example_01_basic_run.py
# End-to-end smoke test
# ── make repo root importable ─────────────────────────────────────────────
import pathlib, sys
sys.path.insert(0, str(pathlib.Path(__file__).resolve().parents[1]))
# ─────────────────────────────────────────────────────────────────────────
import numpy as np
import wfgy_sdk as w
prompt = "Why don't AIs like to take showers?"
rng = np.random.default_rng(0)
G = rng.normal(size=128)
G /= np.linalg.norm(G) # unit vector
I = G + rng.normal(scale=0.05, size=128) # small perturbation
G = rng.normal(size=128); G /= np.linalg.norm(G)
I = G + rng.normal(scale=0.05, size=128)
logits = rng.normal(size=32000)