From 105099d83ef67712b305f962de9551d0e97165bd Mon Sep 17 00:00:00 2001 From: "Henry Dieckhaus (dieckhau)" Date: Sun, 17 May 2026 15:31:43 -0400 Subject: [PATCH] Fixed bug causing yaml file loading to fail on design model 002. --- hbdesigner/inference/inference_hbdesigner.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/hbdesigner/inference/inference_hbdesigner.py b/hbdesigner/inference/inference_hbdesigner.py index 07b0b41..cb486f7 100755 --- a/hbdesigner/inference/inference_hbdesigner.py +++ b/hbdesigner/inference/inference_hbdesigner.py @@ -246,7 +246,7 @@ def validate_inputs(self): # Retrieve model weights and configurations pack_cfg_name = "pack_cpu.yaml" if self.opts.cpu else "pack.yaml" design_cfg_name = ( - f"{self.opts.design_model}_cpu.yaml" if self.opts.cpu else f"{self.opts.design_model}.yaml" + f"design_020_cpu.yaml" if self.opts.cpu else f"design_020.yaml" ) self.opts.pack_cfg = os.path.join(Path(__file__).parents[2], f"model_weights/{pack_cfg_name}") # Override model weight paths if provided by user, otherwise use defaults