Skip to content

Commit be16666

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 150927d commit be16666

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

tests/conftest.py

+2
Original file line numberDiff line numberDiff line change
@@ -95,6 +95,8 @@ def override_model_loading(monkeypatch):
9595
monkeypatch.setattr(safetensors.torch, "load_file", load_stripped_model)
9696
monkeypatch.setattr(gguf_loaders, "gguf_sd_loader", load_stripped_model)
9797
monkeypatch.setattr("invokeai.backend.model_manager.config.gguf_sd_loader", load_stripped_model)
98+
monkeypatch.setattr("invokeai.backend.model_manager.util.model_util.gguf_sd_loader", load_stripped_model)
99+
monkeypatch.setattr("invokeai.backend.model_manager.legacy_probe.gguf_sd_loader", load_stripped_model)
98100

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

0 commit comments

Comments
 (0)