diff --git a/recipes/configs/gemma/evaluation.yaml b/recipes/configs/gemma/evaluation.yaml index fa0e0eebc6..2ff8f78546 100644 --- a/recipes/configs/gemma/evaluation.yaml +++ b/recipes/configs/gemma/evaluation.yaml @@ -15,17 +15,16 @@ checkpointer: model-00001-of-00002.safetensors, model-00002-of-00002.safetensors, ] - #recipe_checkpoint: null output_dir: ./ # Not needed model_type: GEMMA # Tokenizer tokenizer: _component_: torchtune.models.gemma.gemma_tokenizer - path: /tmp/gemma-2b//tokenizer.model + path: /tmp/gemma-2b/tokenizer.model # Environment -device: gpu +device: cuda dtype: bf16 seed: 1234 # It is not recommended to change this seed, b/c it matches EleutherAI's default seed