Skip to content

Enable model loading options when querying an on-disk index#113

Merged
luciaquirke merged 6 commits intomainfrom
query-model
Jan 8, 2026
Merged

Enable model loading options when querying an on-disk index#113
luciaquirke merged 6 commits intomainfrom
query-model

Conversation

@luciaquirke
Copy link
Collaborator

@luciaquirke luciaquirke commented Jan 7, 2026

  • Enable reverse mode queries (lowest influence)
  • Enable device map auto and other model loading options in query
  • Fix types
  • Add two missing dependencies (they were also transitive deps)

"""Field to use for the query."""

unit_norm: bool = False
unit_norm: bool = True
Copy link
Collaborator Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

This was one of the two strongest ablations in TrackStar, I think it makes sense to enable it by default

@luciaquirke luciaquirke changed the title Enable device_map="auto" when querying an on-disk index Enable model loading options when querying an on-disk index Jan 7, 2026
@luciaquirke luciaquirke force-pushed the query-model branch 2 times, most recently from 4c4f585 to c55b5b7 Compare January 8, 2026 04:53
@luciaquirke luciaquirke requested a review from LouisYRYJ January 8, 2026 04:56
@luciaquirke luciaquirke merged commit 8399780 into main Jan 8, 2026
5 checks passed
@luciaquirke luciaquirke deleted the query-model branch January 13, 2026 23:20
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant