Skip to content

Commit cd34d7c

Browse files
committed
corrected comment indicating source of checkpoint
1 parent 1ea009e commit cd34d7c

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

scripts/checkpoint_conversion/numerical_tests_example.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -108,17 +108,17 @@ def forward_tt(config_path, checkpoint_path, test_set):
108108
if __name__ == "__main__":
109109
# hf params
110110
hf_model_name = "meta-llama/Meta-Llama-3-8B"
111-
hf_model_path = "outputs/test_checkpoint/step-0-tohf" # safetensors checkpoint from convert_from_hf.py
112-
# safetensors checkpoint from convert_from_hf.py but without using sd_adapter's permute
111+
hf_model_path = "outputs/test_checkpoint/step-0-tohf" # safetensors checkpoint from convert_to_hf.py
112+
# safetensors checkpoint from convert_to_hf.py but without using sd_adapter's permute
113113
hf_model_path_no_perm = "outputs/test_checkpoint/step-0-tohfnoperm"
114114

115115
# tt params
116116
config_path = "torchtitan/models/llama3/train_configs/llama3_8b.toml"
117117
baseline_checkpoint_path = "outputs/test_checkpoint/step-0-fromllama" # dcp checkpoint from convert_from_llama.py
118118
checkpoint_path = (
119-
"outputs/test_checkpoint/step-0-fromhf" # dcp checkpoint from convert_to_hf.py
119+
"outputs/test_checkpoint/step-0-fromhf" # dcp checkpoint from convert_from_hf.py
120120
)
121-
# dcp checkpoint from convert_to_hf.py without using sd_adapter's permute
121+
# dcp checkpoint from convert_from_hf.py without using sd_adapter's permute
122122
checkpoint_path_no_perm = "outputs/test_checkpoint/step-0-fromhfnoperm"
123123

124124
# test params

0 commit comments

Comments
 (0)