Upload install_musubi.sh

install_musubi.sh

@@ -5,6 +5,26 @@ echo "==========================================================="
 echo "        MusubiPrime: Iniciando Instalação Otimizada        "
 echo "==========================================================="
 
+# --- 1. Instalando Jupyter Notebook e VS Code ---
+echo "INFO: Instalando Jupyter Notebook..."
+pip install jupyter
+
+echo "INFO: Instalando VS Code Server..."
+curl -fsSL https://code-server.dev/install.sh | sh
+
+# --- 2. Verificação do Ambiente ---
+echo "INFO: Verificando a versão do Python..."
+python3 --version
+# O ambiente RunPod já possui uma versão adequada do Python (3.11).
+
+# --- 3. Clonando o Repositório Musubi Trainer ---
+echo "INFO: Clonando o repositório Musubi Trainer..."
+if [ -d "musubi-tuner" ]; then
+    echo "WARN: O diretório 'musubi-tuner' já existe. Pulando a clonagem."
+else
+    git clone https://github.com/kohya-ss/musubi худуд da instalação otimizada        "
+echo "==========================================================="
+
 # --- 1. Verificação do Ambiente ---
 echo "INFO: Verificando a versão do Python..."
 python3 --version
@@ -34,8 +54,6 @@ echo "INFO: Atualizando pip..."
 pip install --upgrade pip
 
 echo "INFO: Atualizando PyTorch para uma versão estável compatível com CUDA 12.x..."
-# Conforme README.md e Issue #273, uma versão mais recente do PyTorch é necessária.
-# Usamos o índice cu121, que é o correto para o ambiente CUDA 12.4.1 da imagem.
 pip install --upgrade torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu121
 
 echo "INFO: Instalando dependências do requirements.txt..."
@@ -49,21 +67,16 @@ pip install bitsandbytes --upgrade --pre
 echo "INFO: Instalando dependências opcionais (tensorboard, matplotlib)..."
 pip install tensorboard matplotlib
 
-# --- 7. Instalação Opcional de Aceleradores de INFERÊNCIA ---
-# Estes são para INFERÊNCIA e NÃO são necessários para o TREINAMENTO.
-# Eles são conhecidos por causar problemas se usados incorretamente durante o treinamento (Issue #92).
-# Descomente as linhas abaixo apenas se você precisar de inferência acelerada.
-
-# echo "INFO: Comandos para instalação opcional de aceleradores de inferência:"
-# echo "      (Descomente no script se necessário)"
-#
-# # FlashAttention (para GPUs Ampere e mais recentes)
-# pip install flash-attn --no-build-isolation
-#
-# # SageAttention (requer Triton)
-# # pip install -U "triton-pre-release" -f https://storage.googleapis.com/triton-nightly/index.html
-# # git clone https://github.com/thu-ml/SageAttention.git
-# # cd SageAttention && pip install . && cd ..
+# --- 7. Configurando e Iniciando Jupyter Notebook e VS Code Server ---
+echo "INFO: Configurando e iniciando Jupyter Notebook na porta 8888..."
+jupyter notebook --ip=0.0.0.0 --port=8888 --allow-root --no-browser &
+
+echo "INFO: Configurando e iniciando VS Code Server na porta 8818..."
+code-server --bind-addr 0.0.0.0:8818 --auth none &
+
+# --- 8. Mantendo o container ativo ---
+echo "INFO: Mantendo o container ativo..."
+sleep infinity
 
 echo "==========================================================="
 echo "      Instalação do Musubi Trainer Concluída com Sucesso!    "