Skip to content

Commit 05ae4e2

Browse files
authored
Stop checking for unet/model.onnx when a model_index.json is detected (#4132)
## What type of PR is this? (check all applicable) - [ ] Refactor - [ ] Feature - [x] Bug Fix - [ ] Optimization - [ ] Documentation Update - [ ] Community Node Submission ## Have you discussed this change with the InvokeAI team? - [x] Yes - [ ] No, because: ## Have you updated all relevant documentation? - [ ] Yes - [ ] No ## Description ## Related Tickets & Documents <!-- For pull requests that relate or close an issue, please include them below. For example having the text: "closes #1234" would connect the current pull request to issue 1234. And when we merge the pull request, Github will automatically close the issue. --> - Related Issue # - Closes # ## QA Instructions, Screenshots, Recordings <!-- Please provide steps on how to test changes, any hardware or software specifications as well as any other pertinent information. --> ## Added/updated tests? - [ ] Yes - [ ] No : _please replace this line with details on why tests have not been included_ ## [optional] Are there any post deployment tasks we need to perform?
2 parents 9091e19 + f06fee4 commit 05ae4e2

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

invokeai/backend/install/model_install_backend.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -303,7 +303,7 @@ def _install_repo(self, repo_id: str) -> AddModelResult:
303303

304304
with TemporaryDirectory(dir=self.config.models_path) as staging:
305305
staging = Path(staging)
306-
if "model_index.json" in files and "unet/model.onnx" not in files:
306+
if "model_index.json" in files:
307307
location = self._download_hf_pipeline(repo_id, staging) # pipeline
308308
elif "unet/model.onnx" in files:
309309
location = self._download_hf_model(repo_id, files, staging)

0 commit comments

Comments
 (0)