Skip to content

Commit

Permalink
mkdir in convert_lit_checkpoint
Browse files Browse the repository at this point in the history
Co-authored-by: Carlos Mocholí <carlossmocholi@gmail.com>
  • Loading branch information
rasbt and carmocca authored Mar 4, 2024
1 parent 4b86dab commit cc17394
Showing 1 changed file with 2 additions and 0 deletions.
2 changes: 2 additions & 0 deletions scripts/convert_lit_checkpoint.py
Original file line number Diff line number Diff line change
Expand Up @@ -247,6 +247,8 @@ def check_conversion_supported(lit_weights: Dict[str, torch.Tensor]) -> None:
def convert_lit_checkpoint(checkpoint_path: Path, output_path: Path, config_path: Path) -> None:
config = Config.from_json(config_path)

output_path.parent.mkdir(parents=True, exist_ok=True)

if "falcon" in config.name:
copy_fn = partial(copy_weights_falcon, config.name)
elif config._mlp_class in ("LLaMAMLP", "GemmaMLP", "LLaMAMoE"):
Expand Down

0 comments on commit cc17394

Please sign in to comment.