Yersel commited on
Commit
104faa8
·
1 Parent(s): 1a3530a

fix model_name

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -10,7 +10,7 @@ login(token)
10
 
11
  model_name = 'meta-llama/Meta-Llama-3.1-8B-Instruct'
12
  model = AutoModelForCausalLM.from_pretrained(model_name, torch_dtype = torch.bfloat16)
13
- tokenizer = AutoTokenizer.from_pretrained(madel_name)
14
 
15
  if torch.cuda.is_available():
16
  device = torch.device('cuda')
 
10
 
11
  model_name = 'meta-llama/Meta-Llama-3.1-8B-Instruct'
12
  model = AutoModelForCausalLM.from_pretrained(model_name, torch_dtype = torch.bfloat16)
13
+ tokenizer = AutoTokenizer.from_pretrained(model_name)
14
 
15
  if torch.cuda.is_available():
16
  device = torch.device('cuda')