Inference Unlimited

Poradnik: Jak uruchomić Stable Beluga na komputerze z 32GB RAM

Stable Beluga to nowoczesne narzędzie do generowania obrazów oparte na technologii AI. W tym poradniku pokażemy, jak uruchomić Stable Beluga na komputerze z 32GB RAM, krok po kroku.

Wymagania wstępne

Przed rozpoczęciem instalacji upewnij się, że Twój system spełnia następujące wymagania:

Instalacja zależności

Na systemie Windows

  1. Zainstaluj sterowniki dla karty graficznej:

    • Pobierz najnowsze sterowniki dla Twojej karty graficznej z oficjalnej strony NVIDIA.
    • Wykonaj instalację zgodnie z instrukcjami producenta.
  2. Zainstaluj Python:

    • Pobierz i zainstaluj Python 3.8 lub nowszy z oficjalnej strony python.org.
    • Upewnij się, że zaznaczysz opcję "Add Python to PATH" podczas instalacji.
  3. Zainstaluj Git:

    • Pobierz i zainstaluj Git z oficjalnej strony git-scm.com.

Na systemie Linux (Ubuntu 20.04 LTS)

  1. Zaktualizuj system:

    sudo apt update && sudo apt upgrade -y
    
  2. Zainstaluj sterowniki NVIDIA:

    sudo apt install nvidia-driver-470
    sudo reboot
    
  3. Zainstaluj Python i Git:

    sudo apt install python3 python3-pip git
    

Klonowanie repozytorium Stable Beluga

  1. Otwórz terminal (lub wiersz polecenia w Windowsie).
  2. Wykonaj następujące polecenie:
    git clone https://github.com/stability-ai/stable-beluga.git
    cd stable-beluga
    

Instalacja zależności Python

  1. Utwórz i aktywuj wirtualne środowisko:

    python -m venv venv
    source venv/bin/activate  # Na Windowsie: venv\Scripts\activate
    
  2. Zainstaluj wymagane pakiety:

    pip install -r requirements.txt
    

Konfiguracja Stable Beluga

  1. Skopiuj plik konfiguracyjny:

    cp config.example.yaml config.yaml
    
  2. Edytuj plik config.yaml w edytorze tekstu, np. VS Code:

    model:
      name: "stable-beluga"
      path: "models/stable-beluga.safetensors"
    device: "cuda"  # Użyj "cpu", jeśli nie masz karty graficznej
    precision: "fp16"  # Możesz zmienić na "fp32", jeśli masz wystarczająco dużo pamięci
    

Pobranie modelu

  1. Pobierz model Stable Beluga z oficjalnego repozytorium:
    wget https://example.com/models/stable-beluga.safetensors -P models/
    
    (Zastąp URL adresem aktualnego modelu.)

Uruchomienie Stable Beluga

  1. Uruchom skrypt główny:

    python main.py
    
  2. W przypadku problemów z pamięcią, spróbuj zmniejszyć rozmiar batcha w pliku config.yaml:

    batch_size: 1  # Domyślnie 4, zmniejsz, jeśli potrzebujesz
    

Przykłady użycia

Generowanie obrazu

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")

Generowanie serii obrazów

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")

Optymalizacja pamięci

Jeśli napotkasz problemy z pamięcią, spróbuj następujących rozwiązań:

  1. Zmniejsz rozmiar batcha:

    batch_size: 1
    
  2. Użyj precyzji fp32:

    precision: "fp32"
    
  3. Wyłącz gradienty:

    enable_gradients: false
    

Rozwiązywanie problemów

Błąd pamięci GPU

Jeśli otrzymasz błąd pamięci GPU, spróbuj:

  1. Zmniejszyć rozdzielczość wyjściową:

    output_size: [512, 512]  # Zmniejsz do [256, 256], jeśli potrzebujesz
    
  2. Użyć mniejszego modelu:

    model:
      name: "stable-beluga-small"
    

Błąd zależności

Jeśli napotkasz problemy z zależnościami, spróbuj:

  1. Zaktualizować pip:

    pip install --upgrade pip
    
  2. Zainstalować pakiety ręcznie:

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

Podsumowanie

W tym poradniku pokazaliśmy, jak uruchomić Stable Beluga na komputerze z 32GB RAM. Pamiętaj, że optymalizacja pamięci może być konieczna w zależności od Twojego sprzętu. Dzięki tym krokom powinieneś być w stanie generować wysokiej jakości obrazy za pomocą Stable Beluga.

Jeśli masz dodatkowe pytania lub napotkasz problemy, odwiedź oficjalne forum Stable Beluga lub repozytorium GitHub.

Język: PL | Wyświetlenia: 8

← Powrót do listy artykułów