Skip to content

Commit 75df37c

Browse files
authored
Merge pull request #21 from anfredette/new-blis-file
Make model name matching case insensitive
2 parents 39a5420 + 86ae5e2 commit 75df37c

File tree

2 files changed

+615
-171
lines changed

2 files changed

+615
-171
lines changed

backend/src/recommendation/capacity_planner.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -92,9 +92,9 @@ def plan_capacity(
9292
)
9393
return None
9494

95-
# Filter to only this model
95+
# Filter to only this model (case-insensitive match)
9696
model_configs = [
97-
bench for bench in matching_configs if bench.model_hf_repo == model.model_id
97+
bench for bench in matching_configs if bench.model_hf_repo.lower() == model.model_id.lower()
9898
]
9999

100100
if not model_configs:

0 commit comments

Comments
 (0)