Fabian
45 lines
1.3 KiB
Python
45 lines
1.3 KiB
Python
"""Quick status: is ComfyUI still running and what's the log say?"""
|
|
import paramiko
|
|
|
|
k = paramiko.Ed25519Key.from_private_key_file(r'C:\Users\fabia\.ssh\id_ed25519')
|
|
c = paramiko.SSHClient()
|
|
c.set_missing_host_key_policy(paramiko.AutoAddPolicy())
|
|
c.connect('192.168.178.150', username='fabian', pkey=k, timeout=10)
|
|
|
|
sftp = c.open_sftp()
|
|
|
|
# Read full log
|
|
with sftp.open('/tmp/comfyui.log', 'r') as f:
|
|
log = f.read().decode(errors='replace')
|
|
|
|
sftp.close()
|
|
|
|
lines = log.split('\n')
|
|
print(f"Total lines: {len(lines)}")
|
|
print()
|
|
|
|
# Show only meaningful lines
|
|
for line in lines:
|
|
s = line.strip()
|
|
if not s:
|
|
continue
|
|
if 'FETCH ComfyRegistry' in s or 'All startup tasks' in s or 'FETCH DATA' in s:
|
|
continue
|
|
print(s)
|
|
|
|
# Check process + GPU
|
|
chan = c.get_transport().open_session()
|
|
chan.settimeout(10)
|
|
chan.exec_command('/bin/bash -c "echo; echo === PROCESS ===; ps aux | grep python3 | grep -v grep; echo; echo === GPU ===; rocm-smi 2>/dev/null | head -12; echo; echo === OUTPUT ===; ls -la ~/ComfyUI/output/ 2>/dev/null; echo; echo === QUEUE ===; curl -s http://127.0.0.1:8188/queue 2>/dev/null"')
|
|
out = b""
|
|
while True:
|
|
try:
|
|
chunk = chan.recv(65536)
|
|
if not chunk: break
|
|
out += chunk
|
|
except: break
|
|
chan.close()
|
|
print(out.decode(errors='replace'))
|
|
|
|
c.close()
|