Skip to content

Commit

Permalink
fix paths
Browse files Browse the repository at this point in the history
  • Loading branch information
iankur committed Sep 1, 2024
1 parent 6b05132 commit a98e46d
Show file tree
Hide file tree
Showing 2 changed files with 5 additions and 5 deletions.
6 changes: 3 additions & 3 deletions recipes/config/eleuther_evaluation.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -20,16 +20,16 @@ model:

checkpointer:
_component_: vqllm.utils.checkpointer.FullModelMetaCheckpointer
checkpoint_dir: /home/ubuntu/vqLLM/recipes/ckpts/llama3_8b/original
checkpoint_dir: /home/ubuntu/vqllm/recipes/ckpts/llama3_8b/original
checkpoint_files: ['consolidated.00.pth']
recipe_checkpoint: null
output_dir: /home/ubuntu/vqLLM/recipes/ckpts
output_dir: /home/ubuntu/vqllm/recipes/ckpts
model_type: LLAMA3

# Tokenizer
tokenizer:
_component_: torchtune.models.llama3.llama3_tokenizer
path: /home/ubuntu/vqLLM/recipes/ckpts/llama3_8b/original/tokenizer.model
path: /home/ubuntu/vqllm/recipes/ckpts/llama3_8b/original/tokenizer.model

# Environment
device: cuda
Expand Down
4 changes: 2 additions & 2 deletions recipes/config/llama3_8b_single_device.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -18,7 +18,7 @@
# Tokenizer
tokenizer:
_component_: torchtune.models.llama3.llama3_tokenizer
path: /home/ubuntu/vqLLM/recipes/ckpts/llama3_8b/original/tokenizer.model
path: /home/ubuntu/vqllm/recipes/ckpts/llama3_8b/original/tokenizer.model

# Dataset
dataset:
Expand Down Expand Up @@ -56,7 +56,7 @@ wandb_watch_layers: []

checkpointer:
_component_: vqllm.utils.checkpointer.FullModelMetaCheckpointer
checkpoint_dir: /home/ubuntu/vqLLM/recipes/ckpts/llama3_8b/original
checkpoint_dir: /home/ubuntu/vqllm/recipes/ckpts/llama3_8b/original
checkpoint_files: ['consolidated.00.pth']
recipe_checkpoint: null
output_dir:
Expand Down

0 comments on commit a98e46d

Please sign in to comment.