Uploaded sanitized BC250/ROCm Repository.
This commit is contained in:
@@ -0,0 +1,56 @@
|
||||
#!/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()
|
||||
Reference in New Issue
Block a user