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:
AutoModelForCausalLMztransformersprepare_model_for_int8_trainingzpeftAutoModelForCausalLMWithValueHeadztrl
To ćwiczenie jest częścią kursu
Uczenie przez wzmacnianie z informacją zwrotną od człowieka (RLHF)
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)