Ateeqq commited on
Commit
00df41d
1 Parent(s): c430b6a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -2
app.py CHANGED
@@ -6,8 +6,7 @@ import transformers
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",
 
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",