Skip to content

Commit 105099d

Browse files
author
Henry Dieckhaus (dieckhau)
committed
Fixed bug causing yaml file loading to fail on design model 002.
1 parent b21ae46 commit 105099d

1 file changed

Lines changed: 1 addition & 1 deletion

File tree

hbdesigner/inference/inference_hbdesigner.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -246,7 +246,7 @@ def validate_inputs(self):
246246
# Retrieve model weights and configurations
247247
pack_cfg_name = "pack_cpu.yaml" if self.opts.cpu else "pack.yaml"
248248
design_cfg_name = (
249-
f"{self.opts.design_model}_cpu.yaml" if self.opts.cpu else f"{self.opts.design_model}.yaml"
249+
f"design_020_cpu.yaml" if self.opts.cpu else f"design_020.yaml"
250250
)
251251
self.opts.pack_cfg = os.path.join(Path(__file__).parents[2], f"model_weights/{pack_cfg_name}")
252252
# Override model weight paths if provided by user, otherwise use defaults

0 commit comments

Comments
 (0)