Skip to content
This repository was archived by the owner on Jun 5, 2025. It is now read-only.

Create github action for syncing and exporting vector DB #69

Merged
merged 11 commits into from
Nov 25, 2024

Conversation

yrobla
Copy link
Contributor

@yrobla yrobla commented Nov 22, 2024

Create an action that runs on demand, to pick all data from json files, generate vector embeddings and populate into a DB. Then export the volume into an artifact, that could be used to embed into a docker image for later consumption

return last_hidden.sum(dim=1) / attention_mask.sum(dim=1)[..., None]


def generate_embeddings(text):
Copy link
Contributor

Choose a reason for hiding this comment

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

I am implementing the embedding functionality in the inferencing engine. Once it is done, we can use that and remove this file.

@yrobla yrobla force-pushed the main branch 3 times, most recently from da4ebc1 to c583092 Compare November 25, 2024 09:31
@lukehinds lukehinds merged commit f7baf94 into stacklok:main Nov 25, 2024
Sign up for free to subscribe to this conversation on GitHub. Already have an account? Sign in.
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

4 participants