ShravanHN commited on
Commit
1ea0a52
·
1 Parent(s): d0d017f

changed MODEL TO meta-llama-8b

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -39,8 +39,8 @@ h1 {
39
  """
40
 
41
  # Load the tokenizer and model
42
- tokenizer = AutoTokenizer.from_pretrained("meta-llama/Meta-Llama-3-8B-Instruct")
43
- model = AutoModelForCausalLM.from_pretrained("meta-llama/Meta-Llama-3-8B-Instruct", device_map="auto") # to("cuda:0")
44
  terminators = [
45
  tokenizer.eos_token_id,
46
  tokenizer.convert_tokens_to_ids("<|eot_id|>")
 
39
  """
40
 
41
  # Load the tokenizer and model
42
+ tokenizer = AutoTokenizer.from_pretrained("meta-llama/Meta-Llama-3-8B")
43
+ model = AutoModelForCausalLM.from_pretrained("meta-llama/Meta-Llama-3-8B", device_map="auto") # to("cuda:0")
44
  terminators = [
45
  tokenizer.eos_token_id,
46
  tokenizer.convert_tokens_to_ids("<|eot_id|>")