[diffusion] chore: fetch metadata beside nested lora weights (#36057)
This commit is contained in:
@@ -663,12 +663,16 @@ def maybe_download_lora(
|
|||||||
return os.path.join(source.local_path, selected_file)
|
return os.path.join(source.local_path, selected_file)
|
||||||
|
|
||||||
assert source.repo_id is not None
|
assert source.repo_id is not None
|
||||||
|
allow_patterns = ["*.json", selected_file]
|
||||||
|
selected_parent = os.path.dirname(selected_file)
|
||||||
|
if selected_parent:
|
||||||
|
allow_patterns.insert(1, f"{selected_parent}/*.json")
|
||||||
local_path = maybe_download_model(
|
local_path = maybe_download_model(
|
||||||
source.repo_id,
|
source.repo_id,
|
||||||
local_dir,
|
local_dir,
|
||||||
download,
|
download,
|
||||||
is_lora=True,
|
is_lora=True,
|
||||||
allow_patterns=["*.json", selected_file],
|
allow_patterns=allow_patterns,
|
||||||
revision=resolved_weight.inventory.resolved_revision or source.revision,
|
revision=resolved_weight.inventory.resolved_revision or source.revision,
|
||||||
)
|
)
|
||||||
target = os.path.join(local_path, selected_file)
|
target = os.path.join(local_path, selected_file)
|
||||||
|
|||||||
@@ -176,6 +176,7 @@ def test_lora_tree_url_selects_one_pinned_weight(tmp_path):
|
|||||||
assert download.call_args.kwargs["revision"] == "immutable-sha"
|
assert download.call_args.kwargs["revision"] == "immutable-sha"
|
||||||
assert download.call_args.kwargs["allow_patterns"] == [
|
assert download.call_args.kwargs["allow_patterns"] == [
|
||||||
"*.json",
|
"*.json",
|
||||||
|
"adapters/*.json",
|
||||||
f"adapters/{weight_name}",
|
f"adapters/{weight_name}",
|
||||||
]
|
]
|
||||||
|
|
||||||
|
|||||||
Reference in New Issue
Block a user