cleaned up code for comments

pytorch · Oct 12, 2024 · d1d1233 · d1d1233
1 parent 673f665
commit d1d1233
Showing 1 changed file with 2 additions and 3 deletions.
diff --git 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