mirror of
https://github.com/kuhyx/testsAndMisc-archive.git
synced 2026-07-04 14:23:04 +02:00
11 lines
452 B
Bash
11 lines
452 B
Bash
|
#!/usr/bin/env bash
|
||
|
|
set -e
|
||
|
|
SCRIPT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
|
||
|
|
REPO_ROOT="$(cd "$SCRIPT_DIR/../.." && pwd)"
|
||
|
|
VENV="$REPO_ROOT/.venv"
|
||
|
|
[[ ! -d "$VENV" ]] && python3 -m venv "$VENV"
|
||
|
|
"$VENV/bin/pip" show torch &>/dev/null || "$VENV/bin/pip" install torch transformers numpy scipy bark -q
|
||
|
|
echo "Loading models (this may take a minute on first run)..."
|
||
|
|
export PYTHONUNBUFFERED=1
|
||
|
|
"$VENV/bin/python" "$SCRIPT_DIR/music_generator.py" "$@"
|