A preparação do ambiente a seguir é necessária para instalar o SongGeneration:
- Ambiente PythonRecomendamos o Python 3.8.12 ou superior, e um ambiente virtual pode ser criado via conda
- dependência do sistemaFFmpeg: O FFmpeg precisa ser instalado em sistemas Linux (
yum install ffmpeg
) - Dependências do Python: Uso
pip install -r requirements.txt
Instale o PyTorch e outras dependências necessárias - selecionávelAtenção ao Flash: o Flash Attention pode ser instalado para melhorar o desempenho (requer CUDA 11.8 ou superior)
- peso do modeloDownload do arquivo de pesos do modelo oficialmente disponível na Hugging Face
Uma versão do ComfyUI está disponível para usuários do Windows para simplificar o processo de instalação e é adequada para desenvolvedores não profissionais.
Essa resposta foi extraída do artigoSongGeneration: um modelo de IA de código aberto para gerar músicas e letras de alta qualidadeO