NitroGen-RTX2060-ONNX / scripts /test_game.ps1
patdev's picture
Add RTX 2060 FP16 ONNX DiT runtime and Windows test scripts
4fcca49 verified
Raw
History Blame Contribute Delete
508 Bytes
param([Parameter(Mandatory=$true)][string]$Process, [ValidateSet(4,8,16)][int]$Steps=4)
$ErrorActionPreference="Stop"
$Root=Split-Path -Parent $PSScriptRoot
$Py=Join-Path $Root ".venv\Scripts\python.exe"
$Server=Start-Process -FilePath $Py -ArgumentList @("scripts\serve_onnx.py","--steps",$Steps) -WorkingDirectory $Root -PassThru
try {
Start-Sleep -Seconds 3
& $Py ".vendor\NitroGen\scripts\play.py" --process $Process
} finally {
if (-not $Server.HasExited) { Stop-Process -Id $Server.Id -Force }
}