Skip to content

Commit fb90471

Browse files
ArshErgoncartazio
authored andcommitted
updated url
1 parent 2cbe0a7 commit fb90471

File tree

2 files changed

+2
-2
lines changed

2 files changed

+2
-2
lines changed

src/llm_vm/client.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -14,7 +14,7 @@
1414
default_big_model= conf.settings.big_model
1515

1616
if conf.settings.small_model is not None:
17-
default_small_model= "llama2"
17+
default_small_model= "pythia"
1818
else:
1919
default_small_model = conf.settings.small_model
2020

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="openlm-research/open_llama_3b_v2"
300+
model_uri="meta-llama/Llama-2-70b-hf"
301301

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

0 commit comments

Comments
 (0)