evolving_ideas.infra.local_llm_downloader¶

evolving_ideas.infra.local_llm_downloader

Attributes¶

`logger`
`MODEL_NAME`
`SAVE_DIRECTORY`

Classes¶

LocalLLMDownloader

Downloads and saves the local LLM model and tokenizer.

Module Contents¶

evolving_ideas.infra.local_llm_downloader.logger¶

evolving_ideas.infra.local_llm_downloader.MODEL_NAME = 'sshleifer/tiny-gpt2'¶

evolving_ideas.infra.local_llm_downloader.SAVE_DIRECTORY = './.models/tiny-gpt2'¶

class evolving_ideas.infra.local_llm_downloader.LocalLLMDownloader(model_name: str | None = MODEL_NAME, save_directory: str = SAVE_DIRECTORY)¶

Downloads and saves the local LLM model and tokenizer.

model_name = 'sshleifer/tiny-gpt2'¶

save_directory = './.models/tiny-gpt2'¶

download()¶: Downloads the model and tokenizer, saving them to the specified directory.