Skip to content

Commit

Permalink
Update architecture.py
Browse files Browse the repository at this point in the history
formatting
  • Loading branch information
ElliotStein committed Nov 4, 2024
1 parent 9cf8d3c commit 63afb68
Showing 1 changed file with 1 addition and 2 deletions.
3 changes: 1 addition & 2 deletions mergekit/architecture.py
Original file line number Diff line number Diff line change
Expand Up @@ -572,7 +572,6 @@ def _get_model_parameter_names(repo_id: str) -> list:
return list(ShardedTensorIndex.from_disk(str(model_dir)).tensor_paths.keys())



def _resolve_model_directory(repo_id: str) -> Path:
"""
Resolve the model directory either from a local path, URL, or by downloading from Hugging Face.
Expand Down Expand Up @@ -625,4 +624,4 @@ def _infer_architecture_info(merge_config):
parameter_names=parameter_names,
prefix_tracker=prefix_tracker,
)
]
]

0 comments on commit 63afb68

Please sign in to comment.