Compare commits

2 Commits

Author SHA1 Message Date
Ethanfel fd043f4172 fix: pin transformers<5.0 for EAT model compatibility
EAT remote model code (worstchan/EAT-base_epoch30_finetune_AS2M) is
incompatible with transformers 5.x — missing all_tied_weights_keys
attribute added in the v5 PreTrainedModel API.

Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
2026-04-19 16:11:18 +02:00
Ethanfel 3c3b1d74bb fix: always reinstall torch stack on Windows re-runs
Remove the "skip if torch exists" guard so re-running the setup script
fixes a broken torchvision install.

Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
2026-04-19 16:09:44 +02:00
2 changed files with 12 additions and 14 deletions
+1 -1
View File
@@ -13,7 +13,7 @@ soundfile>=0.12
# or manually: pip install torch torchaudio --index-url https://download.pytorch.org/whl/cu128 # or manually: pip install torch torchaudio --index-url https://download.pytorch.org/whl/cu128
torch>=2.0 torch>=2.0
torchaudio>=2.0 torchaudio>=2.0
transformers>=4.30 transformers>=4.30,<5.0 # EAT remote model code incompatible with transformers 5.x
timm>=0.9 timm>=0.9
# Object detection # Object detection
+8 -10
View File
@@ -22,21 +22,19 @@ if (Test-Path (Join-Path $venvDir "Scripts\python.exe")) {
& "$venvDir\Scripts\Activate.ps1" & "$venvDir\Scripts\Activate.ps1"
# ── PyTorch ─────────────────────────────────────────────── # ── PyTorch ───────────────────────────────────────────────
$hasTorch = python -c "import torch" 2>&1
if ($LASTEXITCODE -eq 0) {
Write-Host "`nPyTorch already installed, skipping." -ForegroundColor Green
} else {
# Detect NVIDIA GPU via nvidia-smi # Detect NVIDIA GPU via nvidia-smi
$hasNvidia = Get-Command nvidia-smi -ErrorAction SilentlyContinue $hasNvidia = Get-Command nvidia-smi -ErrorAction SilentlyContinue
if ($hasNvidia) { if ($hasNvidia) {
Write-Host "`nNVIDIA GPU detected — installing PyTorch with CUDA 12.8..." -ForegroundColor Green $torchIndex = "https://download.pytorch.org/whl/cu128"
pip install torch torchaudio torchvision --index-url https://download.pytorch.org/whl/cu128 Write-Host "`nNVIDIA GPU detected — using CUDA 12.8 PyTorch index" -ForegroundColor Green
} else { } else {
Write-Host "`nNo NVIDIA GPU detected — installing CPU-only PyTorch..." -ForegroundColor Yellow $torchIndex = "https://download.pytorch.org/whl/cpu"
Write-Host "(Audio scanning will work but will be slower without GPU)" -ForegroundColor Yellow Write-Host "`nNo NVIDIA GPU detected — using CPU-only PyTorch index" -ForegroundColor Yellow
pip install torch torchaudio torchvision --index-url https://download.pytorch.org/whl/cpu
}
} }
# Always install/upgrade torch stack from correct index
# (pip install is a no-op if already at the right version)
Write-Host "Installing PyTorch + torchaudio + torchvision..."
pip install torch torchaudio torchvision --index-url $torchIndex
# ── Python deps ─────────────────────────────────────────── # ── Python deps ───────────────────────────────────────────
Write-Host "`nInstalling project dependencies..." Write-Host "`nInstalling project dependencies..."