[pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
This commit is contained in:
parent
1c5d41a5b0
commit
6c08c17267
2 changed files with 12 additions and 5 deletions
|
|
@ -756,7 +756,11 @@ class VideoBackend:
|
|||
entries: dict[str, dict[str, Any]] = {}
|
||||
total = 0
|
||||
|
||||
def add(repo: str, files: list[tuple[str, int]], gguf: Optional[str] = None) -> int:
|
||||
def add(
|
||||
repo: str,
|
||||
files: list[tuple[str, int]],
|
||||
gguf: Optional[str] = None,
|
||||
) -> int:
|
||||
if not files:
|
||||
return 0
|
||||
entry = entries.setdefault(
|
||||
|
|
|
|||
|
|
@ -1861,7 +1861,12 @@ _LTX23_REPO_SIBLINGS = [
|
|||
|
||||
def _plan_api(monkeypatch, repos):
|
||||
class _Api:
|
||||
def model_info(self, repo_id, files_metadata = False, token = None):
|
||||
def model_info(
|
||||
self,
|
||||
repo_id,
|
||||
files_metadata = False,
|
||||
token = None,
|
||||
):
|
||||
return _PlanInfo(repos[repo_id])
|
||||
|
||||
monkeypatch.setattr("huggingface_hub.HfApi", lambda *a, **k: _Api())
|
||||
|
|
@ -1894,9 +1899,7 @@ def test_download_plan_narrows_an_ltx23_pick_and_stages_its_extras(monkeypatch):
|
|||
assert ckpt["gguf_filename"] == "ltx-2.3-22b-distilled.gguf"
|
||||
assert "vae/ltx-2.3-22b-distilled_video_vae.safetensors" in ckpt["files"]
|
||||
assert "vae/ltx-2.3-22b-distilled_audio_vae.safetensors" in ckpt["files"]
|
||||
assert (
|
||||
"text_encoders/ltx-2.3-22b-distilled_embeddings_connectors.safetensors" in ckpt["files"]
|
||||
)
|
||||
assert "text_encoders/ltx-2.3-22b-distilled_embeddings_connectors.safetensors" in ckpt["files"]
|
||||
# The other variant's companions are not this checkpoint's.
|
||||
assert "vae/ltx-2.3-22b-dev_video_vae.safetensors" not in ckpt["files"]
|
||||
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue