Fabian
25 lines
962 B
Python
25 lines
962 B
Python
#!/usr/bin/env python3
|
|
"""Step 1: Kill stuck ComfyUI and check flags."""
|
|
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=60):
|
|
_, stdout, stderr = ssh.exec_command(cmd, timeout=timeout)
|
|
out = stdout.read().decode()
|
|
err = stderr.read().decode()
|
|
print(out.strip() if out.strip() else "")
|
|
if err.strip():
|
|
for l in err.strip().split('\n')[-5:]:
|
|
print(f"STDERR: {l}")
|
|
|
|
print("=== Kill stuck ===")
|
|
run("pkill -f 'python3 main.py' 2>/dev/null; sleep 2; pkill -9 -f 'python3 main.py' 2>/dev/null; sleep 1; echo killed")
|
|
|
|
print("\n=== Check flags ===")
|
|
run("bash -c 'source /home/fabian/comfyui-env/bin/activate && cd /home/fabian/ComfyUI && python3 main.py --help 2>&1 | grep -i -E \"vae|fp16|fp32|force|cpu|novram|lowvram\"'")
|
|
|
|
ssh.close()
|
|
print("\nDone.")
|