anasmkh commited on
Commit
ece361c
·
verified ·
1 Parent(s): 24d5f39

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -22,8 +22,8 @@ llm = HuggingFaceLLM(
22
  max_new_tokens=256,
23
  generate_kwargs={"temperature": 0.1, "do_sample": True},
24
  system_prompt=system_prompt,
25
- tokenizer_name="meta-llama/Meta-Llama-3-8B",
26
- model_name="meta-llama/Meta-Llama-3-8B",
27
  device_map="auto",
28
  model_kwargs={"torch_dtype": torch.float16 }
29
  )
 
22
  max_new_tokens=256,
23
  generate_kwargs={"temperature": 0.1, "do_sample": True},
24
  system_prompt=system_prompt,
25
+ tokenizer_name="anasmkh/new_customized_llama3.1_8b",
26
+ model_name="anasmkh/new_customized_llama3.1_8b",
27
  device_map="auto",
28
  model_kwargs={"torch_dtype": torch.float16 }
29
  )