Kom igångKom igång gratis

Spara anpassade recept

Kunden har nu bett om en ändring i kraven. Den här gången vill de öka antalet parametrar och använda Llama 3.2-modellen med 3B parametrar. Du gör den här ändringen i din dictionary och sparar den sedan som en YAML-fil.

Biblioteket yaml är redan importerat.

Den här övningen är en del av kursen

Finjustering med Llama 3

Visa kurs

Övningsinstruktioner

  • Ange det nya modellkravet, modellen torchtune.models.llama3_2.llama3_2_3b, i din dictionary.
  • Spara kraven som en YAML-fil med namnet custom_recipe.yaml.

Interaktiv övning med praktiskt arbete

Testa den här övningen genom att slutföra den här exempelkoden.

config_dict = {
    # Update the model
    ____,
    "batch_size": 8,
    "device": "cuda",
    "optimizer": {"_component_": "bitsandbytes.optim.PagedAdamW8bit", "lr": 3e-05},
    "dataset": {"_component_": "custom_dataset"},
    "output_dir": "/tmp/finetune_results"
}

# Save the updated configuration to a new YAML file
with open("custom_recipe.yaml", "w") as yaml_file:
    ____
Redigera och kör kod