Co-Authored-By: Claude Sonnet 5.5 <noreply@anthropic.com> Claude-Session: https://claude.ai/code/session_01Tw9fjV3UWsHHjJiAXix29B
16 lines
1.1 KiB
Bash
Executable File
16 lines
1.1 KiB
Bash
Executable File
#!/bin/sh
|
|
# Test environment for a machine with no GPU (a cloud session): CPU torch, pytest, a ComfyUI checkout.
|
|
# Safe to run again; each step is skipped when already there. Prints the command that runs the suite.
|
|
set -e
|
|
COMFY=${COMFYUI_ROOT:-$HOME/ComfyUI}
|
|
# The three together, from the CPU index: ComfyUI's requirements would otherwise pull a CUDA torchvision that no longer matches.
|
|
python3 -c "import torch, torchvision" 2>/dev/null || pip install --index-url https://download.pytorch.org/whl/cpu torch torchvision torchaudio
|
|
python3 -c "import pytest" 2>/dev/null || pip install pytest
|
|
[ -f "$COMFY/comfy/supported_models.py" ] || git clone --depth 1 https://github.com/comfyanonymous/ComfyUI.git "$COMFY"
|
|
pip install -r "$COMFY/requirements.txt"
|
|
# Optional: the Shot camera's tagger (without it those tests skip and the camera falls back to word rules).
|
|
python3 -c "import spacy" 2>/dev/null || pip install spacy || true
|
|
python3 -c "import spacy; spacy.load('en_core_web_sm')" 2>/dev/null || python3 -m spacy download en_core_web_sm || true
|
|
HERE=$(cd "$(dirname "$0")" && pwd)
|
|
echo "COMFYUI_ROOT=$COMFY PYTHONPATH=$HERE python3 -m pytest -p cpuforce -q"
|