We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent fb90471 commit b2e0cdaCopy full SHA for b2e0cda
src/llm_vm/onsite_llm.py
@@ -297,7 +297,7 @@ class Small_Local_LLama(Base_Onsite_LLM):
297
tokenizer_loader: Loads the tokenizer into memory
298
generate: Generates a response from a given prompt with the loaded LLM and tokenizer
299
"""
300
- model_uri="meta-llama/Llama-2-70b-hf"
+ model_uri="meta-llama/Llama-2-7b"
301
302
def model_loader(self):
303
return Llama2ForCausalLM.from_pretrained(self.model_uri)
0 commit comments