Change llama_checkpoint_path, can't be inside of ckpts folder

Signed-off-by: Daniel Korzekwa <[email protected]>

Author: danielkorzekwa

tests/experimental/torch/_compress/nas/plugins/test_nas_convert.py

@@ -86,8 +86,7 @@ def run_nas_convert(
     # The inputs for the nas.convert() step.
     #
     puzzle_dir = tmp_path
-    # TODO: change it to "ckpts/llama" once the conversion script is fixed (internal NVidia modelopt bug: issues/17)
-    llama_checkpoint_path = puzzle_dir / "ckpts/teacher"
+    llama_checkpoint_path = puzzle_dir / "input_model/llama"
     dataset_path = puzzle_dir / "dummy_dataset"
     hydra_config_dir = project_root_path / "tests/experimental/torch/_compress/resources/configs"
     hydra_config_name = "Llama-3_1-8B"