From d1d12338ec7cf03444611910b9b5761e6698d52d Mon Sep 17 00:00:00 2001
From: malinjawi <linjawi.mohammad@gmail.com>
Date: Sat, 12 Oct 2024 22:41:05 +0300
Subject: [PATCH] cleaned up code for comments

---
 recipes/configs/gemma/evaluation.yaml | 5 ++---
 1 file changed, 2 insertions(+), 3 deletions(-)

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