ghostai1 commited on
Commit
f558cea
·
verified ·
1 Parent(s): 8450fd0

Create install.sh

Browse files
Files changed (1) hide show
  1. install.sh +81 -0
install.sh ADDED
@@ -0,0 +1,81 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ #!/bin/bash
2
+
3
+ # GhostAI Music Generator Full Setup Script
4
+ # Target: Ubuntu + CUDA 12.1 + PyTorch 2.1.0 Environment
5
+
6
+ # Exit on first error
7
+ set -e
8
+
9
+ # ===[ Environment Setup ]===
10
+ PROJECT_DIR="$HOME/ghostai_musicgen"
11
+ VENV_DIR="$PROJECT_DIR/venv"
12
+ PYTHON_VERSION="python3"
13
+
14
+ echo "=== Checking Python 3 Installation ==="
15
+ if ! command -v $PYTHON_VERSION &> /dev/null; then
16
+ echo "Python 3 not found. Install it with: sudo apt update && sudo apt install python3 python3-pip python3-venv"
17
+ exit 1
18
+ fi
19
+
20
+ echo "=== Checking FFmpeg Installation ==="
21
+ if ! command -v ffmpeg &> /dev/null; then
22
+ echo "Installing FFmpeg..."
23
+ sudo apt update
24
+ sudo apt install -y ffmpeg
25
+ else
26
+ echo "FFmpeg is already installed."
27
+ fi
28
+
29
+ echo "=== Preparing Project Directory ==="
30
+ mkdir -p "$PROJECT_DIR"
31
+ cd "$PROJECT_DIR"
32
+
33
+ if [ -d "$VENV_DIR" ]; then
34
+ echo "Removing existing virtual environment at $VENV_DIR"
35
+ rm -rf "$VENV_DIR"
36
+ fi
37
+
38
+ echo "Creating new virtual environment..."
39
+ $PYTHON_VERSION -m venv "$VENV_DIR"
40
+ source "$VENV_DIR/bin/activate"
41
+
42
+ echo "=== Upgrading Pip ==="
43
+ pip install --upgrade pip
44
+
45
+ # ===[ Install Core Dependencies ]===
46
+ echo "Installing NumPy 1.26.4 (latest compatible)"
47
+ pip install numpy==1.26.4
48
+
49
+ echo "Installing Thinc 8.2.5 (resolves numpy>=2.0.0 dependency conflicts)"
50
+ pip install thinc==8.2.5
51
+
52
+ echo "Installing PyTorch 2.1.0 (CUDA 12.1)"
53
+ pip install torch==2.1.0 torchvision==0.16.0 torchaudio==2.1.0 --index-url https://download.pytorch.org/whl/cu121
54
+
55
+ echo "Installing Audiocraft and Other Dependencies"
56
+ pip install audiocraft==1.3.0 gradio==4.36.1 pydub==0.25.1 psutil==6.1.0 pedalboard==0.9.16 transformers==4.51.3 scipy==1.12.0
57
+
58
+ # ===[ Model Download ]===
59
+ echo "Preparing models directory..."
60
+ mkdir -p models/musicgen-medium
61
+
62
+ echo "Downloading MusicGen-Medium model..."
63
+ python3 -c "from audiocraft.models import MusicGen; MusicGen.get_pretrained('facebook/musicgen-medium')"
64
+
65
+ echo "Moving downloaded model to models/musicgen-medium if available..."
66
+ if [ -d ~/.cache/audiocraft/models--facebook--musicgen-medium ]; then
67
+ mv ~/.cache/audiocraft/models--facebook--musicgen-medium/* models/musicgen-medium/
68
+ else
69
+ echo "Warning: Model not found in cache. Skipping move step."
70
+ fi
71
+
72
+ # ===[ Cleanup ]===
73
+ echo "Deactivating virtual environment"
74
+ deactivate
75
+
76
+ echo "=== Setup Complete ==="
77
+ echo "To activate your environment:"
78
+ echo " source $VENV_DIR/bin/activate"
79
+ echo "To run the application:"
80
+ echo " python $PROJECT_DIR/newmuzak.py"
81
+ echo "Visit: http://localhost:9999 in your browser"