Comece agoraComece grátis

Salvando receitas personalizadas

O cliente agora pediu uma alteração nos requisitos. Desta vez, eles gostariam de aumentar o número de parâmetros e usar o modelo Llama 3.2 com 3B de parâmetros. Você faz essa modificação no seu dicionário e depois salva como um arquivo YAML.

A biblioteca yaml já foi importada.

Este exercicio faz parte do curso

Ajuste Fino com Llama 3

Ver curso

Instruções do exercicio

  • Especifique o novo requisito de modelo, o torchtune.models.llama3_2.llama3_2_3b, no seu dicionário.
  • Salve os requisitos como um arquivo YAML chamado custom_recipe.yaml.

exercicio interativo prático

Tente este exercicio completando este código de exemplo.

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:
    ____
Editar e Executar Código