Uploaded sanitized BC250/ROCm Repository.
This commit is contained in:
Fabian
@@ -0,0 +1,203 @@
|
||||
"""Fix: Remove --novram so ComfyUI actually uses the GPU for compute."""
|
||||
import paramiko, time, json, textwrap
|
||||
|
||||
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=15)
|
||||
|
||||
sftp = c.open_sftp()
|
||||
|
||||
def sh(cmd, timeout=60):
|
||||
chan = c.get_transport().open_session()
|
||||
chan.settimeout(timeout)
|
||||
chan.exec_command('/bin/bash -l -c ' + "'" + cmd.replace("'", "'\\''") + "'")
|
||||
out = b""
|
||||
while True:
|
||||
try:
|
||||
chunk = chan.recv(65536)
|
||||
if not chunk: break
|
||||
out += chunk
|
||||
except: break
|
||||
chan.close()
|
||||
return out.decode(errors='replace').strip()
|
||||
|
||||
# STEP 1: Kill
|
||||
print("STEP 1: Kill")
|
||||
sh('pkill -9 -f "python3.*main.py" 2>/dev/null; sleep 2')
|
||||
print(" Done")
|
||||
|
||||
# STEP 2: Write new launcher - NO --novram, NO --lowvram
|
||||
# ComfyUI sees 7602MB VRAM → will use NORMAL_VRAM mode → GPU compute
|
||||
print("\nSTEP 2: New launcher (NO memory flags = auto GPU)")
|
||||
launcher = textwrap.dedent("""\
|
||||
#!/bin/bash
|
||||
# GPU
|
||||
export HSA_OVERRIDE_GFX_VERSION=10.1.0
|
||||
export HIP_VISIBLE_DEVICES=0
|
||||
export HSA_ENABLE_SDMA=0
|
||||
export HSA_TOOLS_LIB=""
|
||||
export HSA_TOOLS_REPORT_LOAD_FAILURE=0
|
||||
export PYTORCH_HIP_ALLOC_CONF=expandable_segments:False
|
||||
# Threading
|
||||
export OMP_NUM_THREADS=12
|
||||
export MKL_NUM_THREADS=12
|
||||
export OPENBLAS_NUM_THREADS=12
|
||||
# MIOpen - use fast mode, persistent cache
|
||||
export MIOPEN_FIND_MODE=3
|
||||
export MIOPEN_LOG_LEVEL=3
|
||||
|
||||
cd ~/ComfyUI
|
||||
source ~/comfyui-env/bin/activate
|
||||
|
||||
# NO --novram, NO --lowvram = ComfyUI auto-detects 7602MB VRAM = GPU compute
|
||||
# --force-fp16 = half precision saves memory
|
||||
exec python3 main.py \\
|
||||
--listen 0.0.0.0 --port 8188 \\
|
||||
--force-fp16
|
||||
""")
|
||||
with sftp.open('/tmp/run_comfyui.sh', 'w') as f:
|
||||
f.write(launcher)
|
||||
sh('chmod +x /tmp/run_comfyui.sh')
|
||||
print(" Flags: --force-fp16 ONLY (auto VRAM mode)")
|
||||
|
||||
# STEP 3: Start
|
||||
print("\nSTEP 3: Start ComfyUI")
|
||||
sh('rm -f /tmp/comfyui.log; touch /tmp/comfyui.log')
|
||||
sh('nohup /tmp/run_comfyui.sh > /tmp/comfyui.log 2>&1 &')
|
||||
time.sleep(3)
|
||||
pid = sh('pgrep -f "python3.*main.py"')
|
||||
print(f" PID: {pid}")
|
||||
|
||||
# STEP 4: Wait for ready
|
||||
print("\nSTEP 4: Wait for HTTP ready", end='', flush=True)
|
||||
for i in range(120):
|
||||
code = sh('curl -s -o /dev/null -w "%{http_code}" http://127.0.0.1:8188/ 2>/dev/null', timeout=5)
|
||||
if '200' in code:
|
||||
print(f" READY ({i*2}s)")
|
||||
break
|
||||
if i % 10 == 0 and i > 0:
|
||||
try:
|
||||
with sftp.open('/tmp/comfyui.log', 'r') as f:
|
||||
log = f.read().decode(errors='replace')
|
||||
lines = [l.strip() for l in log.split('\n') if l.strip() and 'FETCH' not in l and 'DEPRECATION' not in l]
|
||||
print(f"\n [{i*2}s] {lines[-1][:80] if lines else '...'}", end='', flush=True)
|
||||
except: pass
|
||||
else:
|
||||
print('.', end='', flush=True)
|
||||
time.sleep(2)
|
||||
else:
|
||||
print("\n TIMEOUT!")
|
||||
exit(1)
|
||||
|
||||
# Show VRAM state
|
||||
with sftp.open('/tmp/comfyui.log', 'r') as f:
|
||||
log = f.read().decode(errors='replace')
|
||||
for line in log.split('\n'):
|
||||
s = line.strip()
|
||||
if any(k in s for k in ['vram state', 'Device:', 'Total VRAM', 'VRAM', 'pytorch version']):
|
||||
print(f" {s}")
|
||||
|
||||
# STEP 5: Submit workflow
|
||||
print("\nSTEP 5: Submit workflow")
|
||||
workflow = {
|
||||
"prompt": {
|
||||
"1": {"class_type": "UnetLoaderGGUF", "inputs": {"unet_name": "z_image_turbo-Q5_K_S.gguf"}},
|
||||
"2": {"class_type": "CLIPLoaderGGUF", "inputs": {"clip_name": "Qwen3-4B.i1-Q5_K_S.gguf", "type": "qwen_image"}},
|
||||
"3": {"class_type": "VAELoader", "inputs": {"vae_name": "ae.safetensors"}},
|
||||
"4": {"class_type": "CLIPTextEncode", "inputs": {"text": "A red fox in a snowy forest, photorealistic", "clip": ["2", 0]}},
|
||||
"5": {"class_type": "EmptyLatentImage", "inputs": {"width": 512, "height": 512, "batch_size": 1}},
|
||||
"6": {"class_type": "KSampler", "inputs": {
|
||||
"model": ["1", 0], "positive": ["4", 0], "negative": ["4", 0],
|
||||
"latent_image": ["5", 0], "seed": 42, "steps": 8, "cfg": 1.0,
|
||||
"sampler_name": "euler", "scheduler": "simple", "denoise": 1.0
|
||||
}},
|
||||
"7": {"class_type": "VAEDecode", "inputs": {"samples": ["6", 0], "vae": ["3", 0]}},
|
||||
"8": {"class_type": "SaveImage", "inputs": {"images": ["7", 0], "filename_prefix": "ZImageTurbo_GPU"}}
|
||||
}
|
||||
}
|
||||
with sftp.open('/tmp/wf.json', 'w') as f:
|
||||
f.write(json.dumps(workflow))
|
||||
resp = sh('curl -s -X POST http://127.0.0.1:8188/prompt -H "Content-Type: application/json" -d @/tmp/wf.json', timeout=10)
|
||||
print(f" {resp[:150]}")
|
||||
if 'prompt_id' not in resp:
|
||||
print(" FAILED!")
|
||||
exit(1)
|
||||
|
||||
# STEP 6: Monitor - focus on GPU usage and sampling speed
|
||||
print("\nSTEP 6: Monitor")
|
||||
print(" First step may be slow (MIOpen kernel compilation). Be patient.")
|
||||
t0 = time.time()
|
||||
for i in range(200):
|
||||
elapsed = int(time.time() - t0)
|
||||
|
||||
gpu_temp = sh('cat /sys/class/drm/card0/device/hwmon/hwmon*/temp1_input 2>/dev/null', timeout=5)
|
||||
temp_c = int(gpu_temp) // 1000 if gpu_temp.isdigit() else '?'
|
||||
|
||||
try:
|
||||
with sftp.open('/tmp/comfyui.log', 'r') as f:
|
||||
log = f.read().decode(errors='replace')
|
||||
except: log = ''
|
||||
|
||||
# Find sampling progress and last meaningful line
|
||||
sampling = ''
|
||||
last = ''
|
||||
for line in log.split('\n'):
|
||||
s = line.strip()
|
||||
if '/8' in s and ('it/s' in s or 's/it' in s):
|
||||
sampling = s
|
||||
if s and 'FETCH' not in s and 'startup tasks' not in s and 'DEPRECATION' not in s:
|
||||
last = s
|
||||
|
||||
display = sampling if sampling else last[-100:]
|
||||
print(f" [{elapsed:>4}s] {temp_c}C | {display}")
|
||||
|
||||
# Check output
|
||||
imgs = sh('ls ~/ComfyUI/output/ZImageTurbo_GPU*.png 2>/dev/null', timeout=5)
|
||||
if imgs:
|
||||
# Get timing from log
|
||||
exec_time = ''
|
||||
for line in log.split('\n'):
|
||||
if 'Prompt executed in' in line:
|
||||
exec_time = line.strip()
|
||||
print(f"\n *** IMAGE GENERATED! ***")
|
||||
print(f" File: {imgs}")
|
||||
print(f" {exec_time}")
|
||||
print(f" Wall time: {elapsed}s")
|
||||
|
||||
# Show vram state and model loading details
|
||||
for line in log.split('\n'):
|
||||
s = line.strip()
|
||||
if any(k in s for k in ['loaded completely', 'loaded partially', 'vram state', '/8']):
|
||||
print(f" {s}")
|
||||
break
|
||||
|
||||
# Check queue empty
|
||||
q = sh('curl -s http://127.0.0.1:8188/queue 2>/dev/null', timeout=5)
|
||||
try:
|
||||
qd = json.loads(q)
|
||||
if not qd.get('queue_running') and not qd.get('queue_pending') and elapsed > 30:
|
||||
time.sleep(3)
|
||||
imgs = sh('ls ~/ComfyUI/output/ZImageTurbo_GPU*.png 2>/dev/null', timeout=5)
|
||||
if imgs:
|
||||
print(f"\n *** IMAGE GENERATED: {imgs} ***")
|
||||
else:
|
||||
print(f"\n Queue empty, no image. Log errors:")
|
||||
for line in log.split('\n')[-25:]:
|
||||
if line.strip(): print(f" {line.strip()}")
|
||||
break
|
||||
except: pass
|
||||
|
||||
# Process alive?
|
||||
alive = sh('pgrep -f "python3.*main.py" >/dev/null && echo Y || echo N', timeout=5)
|
||||
if alive == 'N':
|
||||
print(f"\n *** CRASHED ***")
|
||||
for line in log.split('\n')[-30:]:
|
||||
if line.strip(): print(f" {line.strip()}")
|
||||
break
|
||||
|
||||
time.sleep(15)
|
||||
|
||||
sftp.close()
|
||||
c.close()
|
||||
print("\nDone.")
|
||||
Reference in New Issue
Block a user