Inference Unlimited

Guida: Come avviare Stable Beluga su un computer con 32GB di RAM

Stable Beluga è uno strumento moderno per la generazione di immagini basato sulla tecnologia AI. In questa guida, ti mostreremo come avviare Stable Beluga su un computer con 32GB di RAM, passo dopo passo.

Requisiti preliminari

Prima di iniziare l'installazione, assicurati che il tuo sistema soddisfi i seguenti requisiti:

Installazione delle dipendenze

Su sistema Windows

  1. Installa i driver per la scheda grafica:

    • Scarica gli ultimi driver per la tua scheda grafica dal sito ufficiale di NVIDIA.
    • Esegui l'installazione seguendo le istruzioni del produttore.
  2. Installa Python:

    • Scarica e installa Python 3.8 o successivo dal sito ufficiale python.org.
    • Assicurati di selezionare l'opzione "Add Python to PATH" durante l'installazione.
  3. Installa Git:

    • Scarica e installa Git dal sito ufficiale git-scm.com.

Su sistema Linux (Ubuntu 20.04 LTS)

  1. Aggiorna il sistema:

    sudo apt update && sudo apt upgrade -y
    
  2. Installa i driver NVIDIA:

    sudo apt install nvidia-driver-470
    sudo reboot
    
  3. Installa Python e Git:

    sudo apt install python3 python3-pip git
    

Clonazione del repository Stable Beluga

  1. Apri il terminale (o la riga di comando in Windows).
  2. Esegui il seguente comando:
    git clone https://github.com/stability-ai/stable-beluga.git
    cd stable-beluga
    

Installazione delle dipendenze Python

  1. Crea e attiva un ambiente virtuale:

    python -m venv venv
    source venv/bin/activate  # Su Windows: venv\Scripts\activate
    
  2. Installa i pacchetti richiesti:

    pip install -r requirements.txt
    

Configurazione di Stable Beluga

  1. Copia il file di configurazione:

    cp config.example.yaml config.yaml
    
  2. Modifica il file config.yaml in un editor di testo, ad esempio VS Code:

    model:
      name: "stable-beluga"
      path: "models/stable-beluga.safetensors"
    device: "cuda"  # Usa "cpu", se non hai una scheda grafica
    precision: "fp16"  # Puoi cambiare in "fp32", se hai abbastanza memoria
    

Scaricamento del modello

  1. Scarica il modello Stable Beluga dal repository ufficiale:
    wget https://example.com/models/stable-beluga.safetensors -P models/
    
    (Sostituisci l'URL con quello del modello attuale.)

Avvio di Stable Beluga

  1. Esegui lo script principale:

    python main.py
    
  2. In caso di problemi di memoria, prova a ridurre la dimensione del batch nel file config.yaml:

    batch_size: 1  # Predefinito 4, riduci se necessario
    

Esempi di utilizzo

Generazione di un'immagine

from stable_beluga import StableBeluga

model = StableBeluga.from_config("config.yaml")
prompt = "Astronaut riding a horse on Mars"
image = model.generate(prompt)
image.save("output.png")

Generazione di una serie di immagini

prompts = [
    "Sunset over the ocean",
    "Forest in autumn",
    "City skyline at night"
]

for prompt in prompts:
    image = model.generate(prompt)
    image.save(f"{prompt.replace(' ', '_')}.png")

Ottimizzazione della memoria

Se incontri problemi di memoria, prova le seguenti soluzioni:

  1. Riduci la dimensione del batch:

    batch_size: 1
    
  2. Usa la precisione fp32:

    precision: "fp32"
    
  3. Disattiva i gradienti:

    enable_gradients: false
    

Risoluzione dei problemi

Errore di memoria GPU

Se ricevi un errore di memoria GPU, prova:

  1. Ridurre la risoluzione di output:

    output_size: [512, 512]  # Riduci a [256, 256], se necessario
    
  2. Usare un modello più piccolo:

    model:
      name: "stable-beluga-small"
    

Errore di dipendenza

Se incontri problemi con le dipendenze, prova:

  1. Aggiorna pip:

    pip install --upgrade pip
    
  2. Installa i pacchetti manualmente:

    pip install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu113
    

Riassunto

In questa guida, ti abbiamo mostrato come avviare Stable Beluga su un computer con 32GB di RAM. Ricorda che l'ottimizzazione della memoria potrebbe essere necessaria a seconda del tuo hardware. Con questi passaggi, dovresti essere in grado di generare immagini di alta qualità utilizzando Stable Beluga.

Se hai ulteriori domande o incontri problemi, visita il forum ufficiale di Stable Beluga o il repository GitHub.

Język: IT | Wyświetlenia: 7

← Powrót do listy artykułów