diff --git a/mergekit/architecture.py b/mergekit/architecture.py index 5d1004f0..d2c348f7 100644 --- a/mergekit/architecture.py +++ b/mergekit/architecture.py @@ -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. @@ -625,4 +624,4 @@ def _infer_architecture_info(merge_config): parameter_names=parameter_names, prefix_tracker=prefix_tracker, ) - ] + ] \ No newline at end of file