Skip to content

Commit b2e0cda

Browse files
ArshErgoncartazio
authored andcommitted
downgrade the version
1 parent fb90471 commit b2e0cda

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

src/llm_vm/onsite_llm.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -297,7 +297,7 @@ class Small_Local_LLama(Base_Onsite_LLM):
297297
tokenizer_loader: Loads the tokenizer into memory
298298
generate: Generates a response from a given prompt with the loaded LLM and tokenizer
299299
"""
300-
model_uri="meta-llama/Llama-2-70b-hf"
300+
model_uri="meta-llama/Llama-2-7b"
301301

302302
def model_loader(self):
303303
return Llama2ForCausalLM.from_pretrained(self.model_uri)

0 commit comments

Comments
 (0)