Ateeqq commited on
Commit
c430b6a
1 Parent(s): bb15e3d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -6,7 +6,8 @@ import transformers
6
  import torch
7
  from transformers import AutoModelForCausalLM, AutoTokenizer
8
 
9
- model_name = "meta-llama/Meta-Llama-3-8B-Instruct"
 
10
 
11
  pipeline = transformers.pipeline(
12
  "text-generation",
 
6
  import torch
7
  from transformers import AutoModelForCausalLM, AutoTokenizer
8
 
9
+ #model_name = "meta-llama/Meta-Llama-3-8B-Instruct"
10
+ model_name = "meta-llama/Meta-Llama-3.1-8B-Instruct"
11
 
12
  pipeline = transformers.pipeline(
13
  "text-generation",