llm_aided_ocr icon indicating copy to clipboard operation
llm_aided_ocr copied to clipboard

Alternative offline LLMs

Open adnanPBI opened this issue 9 months ago • 2 comments

Hi, Your code used llma2 chat offline LLM. But, I wanted to use alternative offline LLMs such as huggingface's distilbert or roberta or albert. Do you have any suggestion for those LLMs to apply on python base?

adnanPBI avatar May 11 '24 09:05 adnanPBI

i think you would just need to update the download_models function to download the model you want... and if it is not a Llama model then you need to change the get_tokenizer function and lot the right tokenizer

Backendmagier avatar Aug 09 '24 13:08 Backendmagier

get_tokenizer function and lot the right tokenizer

Hi, yes I like to use my own local LLM, but it is a bit tricky. I just need to adjust the tokenizer ?

Shahin-rmz avatar Sep 21 '24 15:09 Shahin-rmz