Fabian
57 lines
1.7 KiB
Python
57 lines
1.7 KiB
Python
#!/usr/bin/env python3
|
|
"""Quick check on ComfyUI status."""
|
|
import paramiko
|
|
ssh = paramiko.SSHClient()
|
|
ssh.set_missing_host_key_policy(paramiko.AutoAddPolicy())
|
|
ssh.connect('192.168.178.150', username='fabian', key_filename=r'C:\Users\fabia\.ssh\id_ed25519')
|
|
|
|
def run(cmd, timeout=30):
|
|
_, stdout, stderr = ssh.exec_command(cmd, timeout=timeout)
|
|
return stdout.read().decode()
|
|
|
|
# Log tail
|
|
print("=== LOG (last 40 lines) ===")
|
|
print(run("tail -40 /home/fabian/comfyui.log 2>/dev/null"))
|
|
|
|
# Process status
|
|
print("=== PROCESS ===")
|
|
print(run("bash -c 'PID=$(pgrep -f \"python3 main.py\" | head -1); "
|
|
"if [ -n \"$PID\" ]; then "
|
|
" ps -p $PID -o pid,%cpu,%mem,nlwp,stat --no-headers; "
|
|
" echo \"LOAD: $(cat /proc/loadavg)\"; "
|
|
"else echo DEAD; fi'"))
|
|
|
|
# rocm-smi
|
|
print("=== GPU ===")
|
|
print(run("HSA_OVERRIDE_GFX_VERSION=10.1.0 rocm-smi 2>/dev/null || echo 'no rocm-smi'"))
|
|
|
|
# Queue
|
|
print("=== QUEUE ===")
|
|
print(run("curl -s http://localhost:8188/queue 2>/dev/null || echo 'no connection'"))
|
|
|
|
# History
|
|
print("=== HISTORY ===")
|
|
hist = run("curl -s http://localhost:8188/history 2>/dev/null || echo 'no connection'")
|
|
import json
|
|
try:
|
|
h = json.loads(hist)
|
|
for pid, info in h.items():
|
|
print(f" Prompt: {pid}")
|
|
print(f" Status: {info.get('status', {})}")
|
|
outputs = info.get('outputs', {})
|
|
if outputs:
|
|
for nid, nout in outputs.items():
|
|
if isinstance(nout, dict):
|
|
for key, val in nout.items():
|
|
print(f" Output node {nid}/{key}: {str(val)[:200]}")
|
|
else:
|
|
print(" No outputs")
|
|
except:
|
|
print(hist[:1000])
|
|
|
|
# Output directory
|
|
print("\n=== OUTPUT FILES ===")
|
|
print(run("ls -lah /home/fabian/ComfyUI/output/ 2>/dev/null"))
|
|
|
|
ssh.close()
|