Zacznij terazZacznij za darmo

Przygotowanie do treningu 8-bitowego

Chciałeś rozpocząć dostrajanie RLHF, ale ciągle napotykałeś błędy braku pamięci. Aby temu zaradzić, postanowiłeś przejść na precyzję 8-bitową, która umożliwia wydajniejsze dostrajanie – z wykorzystaniem biblioteki peft od Hugging Face.

Następujące elementy zostały już zaimportowane:

  • AutoModelForCausalLM z transformers
  • prepare_model_for_int8_training z peft
  • AutoModelForCausalLMWithValueHead z trl

To ćwiczenie jest częścią kursu

Uczenie przez wzmacnianie z informacją zwrotną od człowieka (RLHF)

Zobacz kurs

Instrukcje do ćwiczenia

  • Wczytaj wstępnie wytrenowany model i pamiętaj o uwzględnieniu parametru precyzji 8-bitowej.
  • Użyj funkcji prepare_model_for_int8_training, aby przygotować model do dostrajania opartego na LoRA.
  • Wczytaj model z głowicą wartości (value head) do treningu PPO.

Interaktywne ćwiczenie praktyczne

Spróbuj tego ćwiczenia, uzupełniając ten przykładowy kod.

model_name = "gpt2"  

# Load the model in 8-bit precision
pretrained_model = AutoModelForCausalLM.from_pretrained(
                                                       model_name, 
                                                       ____=True
                                                      )

# Prepare the model for fine-tuning
pretrained_model_8bit = ____(pretrained_model)

# Load the model with a value head
model = ____.from_pretrained(pretrained_model_8bit)
Edytuj i uruchom kod