vrr/WFGY
2
0
Fork 0
mirror of https://github.com/onestardao/WFGY.git synced 2026-04-30 04:29:57 +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 23:57:20 +08:00 • committed by GitHub
parent 242f84a6bc
commit 49732675af
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 6 additions and 6 deletions
Download patch file Download diff file Expand all files Collapse all files

12
examples/example_01_basic_run.py
View file

@ -24,21 +24,18 @@
│ https://www.youtube.com/@OneStarDao │
╰──────────────────────────────────────────────────────────╯
"""
# example_01_basic_run.py
# Basic run with local / remote toggle
import pathlib, sys, numpy as np
sys.path.insert(0, str(pathlib.Path(__file__).resolve().parents[1]))
import wfgy_sdk as w
from wfgy_sdk.evaluator import compare_logits, pretty_print
from wfgy_sdk.visual import plot_histogram # ← histogram helper
# ---------------- toggle ----------------
# ---------- toggle remote / local ------------------------------------------------
use_remote = False
MODEL_ID = "gpt2"
prompt = "Why don't AIs like to take showers?"
# ----------------------------------------
# ---------------------------------------------------------------------------------
if use_remote:
logits_before = w.call_remote_model(prompt, model_id=MODEL_ID)
@ -57,3 +54,6 @@ print("\n=== Example 01 · Basic Run ===")
print(f"Source : {'HF ' + MODEL_ID if use_remote else 'local random'}")
pretty_print(compare_logits(logits_before, logits_after))
print("⚠ Larger LLM → stronger variance drop & higher KL.\n")
# ---------- optional histogram ---------------------------------------------------
plot_histogram(logits_before, logits_after)