Skip to content

Commit 34457fc

Browse files
tests: monkeypatch more references to gguf_sd_loader()
The new util in de9f541 alters import order, breaking some model probe tests. We need to patch more references to `gguf_sd_loader()` to fix em
1 parent 88ea6b5 commit 34457fc

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

tests/conftest.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -97,6 +97,8 @@ def override_model_loading(monkeypatch):
9797
monkeypatch.setattr(safetensors.torch, "load_file", load_stripped_model)
9898
monkeypatch.setattr(gguf_loaders, "gguf_sd_loader", load_stripped_model)
9999
monkeypatch.setattr("invokeai.backend.model_manager.config.gguf_sd_loader", load_stripped_model)
100+
monkeypatch.setattr("invokeai.backend.model_manager.util.model_util.gguf_sd_loader", load_stripped_model)
101+
monkeypatch.setattr("invokeai.backend.model_manager.legacy_probe.gguf_sd_loader", load_stripped_model)
100102

101103
def fake_scan(*args, **kwargs):
102104
return SimpleNamespace(infected_files=0, scan_err=None)

0 commit comments

Comments
 (0)