Claude
527d368c82
Skip fewer attention blocks on early steps and more on late ones, in a straight line up to sla_sparsity. Off at 0, untested on H3. Credit GSRS (arXiv 2610.21129). tools/test_env.sh and tools/cpuforce.py set up the suite on a GPU-less machine. Co-Authored-By: Claude Sonnet 5.5 <noreply@anthropic.com> Claude-Session: https://claude.ai/code/session_0176SHndmQj7T1VtWAWZiNcq
18 lines
448 B
Python
18 lines
448 B
Python
"""pytest plugin: run ComfyUI on the CPU, so the suite works on a machine without a GPU.
|
|
|
|
COMFYUI_ROOT=/path/to/ComfyUI PYTHONPATH=tools python -m pytest -p cpuforce
|
|
"""
|
|
import os
|
|
import sys
|
|
|
|
|
|
def pytest_configure(config):
|
|
root = os.environ.get("COMFYUI_ROOT")
|
|
if root and root not in sys.path:
|
|
sys.path.insert(0, root)
|
|
try:
|
|
from comfy.cli_args import args
|
|
except ImportError:
|
|
return
|
|
args.cpu = True
|