berkus
/
ANE
mirror of https://github.com/maderix/ANE.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

[fix] Dashboard sudo hang fix (upstream PR #20): prevent blocking when password is required for powermetrics

This commit is contained in:
Erik Bray 2026-03-03 14:21:57 +01:00
parent 380237af1f
commit 4ae51e038b
1 changed files with 5 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
training/dashboard.py
View File

@ -142,7 +142,7 @@ def softmax(x):
e = np.exp(x) e = np.exp(x)
return e / np.sum(e) return e / np.sum(e)
def generate_text(W, tok, max_tokens=64, temperature=0.8): def generate_text(W, max_tokens=64, temperature=0.8):
tokenizer = get_tokenizer() tokenizer = get_tokenizer()
if tokenizer is None: if tokenizer is None:
return '[no tokenizer]' return '[no tokenizer]'
@ -244,7 +244,7 @@ def generation_thread():
with S.gen_lock: with S.gen_lock:
S.gen_status = 'idle' S.gen_status = 'idle'
continue continue
text = generate_text(W, get_tokenizer(), max_tokens=64, temperature=0.8) text = generate_text(W, max_tokens=64, temperature=0.8)
with S.gen_lock: with S.gen_lock:
S.gen_text = text S.gen_text = text
S.gen_step = S.step S.gen_step = S.step
@ -672,6 +672,8 @@ def spawn_training(resume=False, steps=10000):
return proc return proc
def spawn_powermetrics(): def spawn_powermetrics():
if not sys.stdin.isatty():
return None
try: try:
proc = subprocess.Popen( proc = subprocess.Popen(
['sudo', 'powermetrics', '--samplers', 'cpu_power,gpu_power,ane_power', '-i', '1000'], ['sudo', 'powermetrics', '--samplers', 'cpu_power,gpu_power,ane_power', '-i', '1000'],
@ -851,7 +853,7 @@ def main():
try: try:
W = load_weights_from_ckpt(CKPT_PATH) W = load_weights_from_ckpt(CKPT_PATH)
if W: if W:
text = generate_text(W, get_tokenizer(), max_tokens=64, temperature=0.8) text = generate_text(W, max_tokens=64, temperature=0.8)
with S.gen_lock: with S.gen_lock:
S.gen_text = text S.gen_text = text
S.gen_step = S.step S.gen_step = S.step