Abhinay45 commited on
Commit
0f04907
·
verified ·
1 Parent(s): bb15e3d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -6,7 +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
 
11
  pipeline = transformers.pipeline(
12
  "text-generation",
 
6
  import torch
7
  from transformers import AutoModelForCausalLM, AutoTokenizer
8
 
9
+ model_name = "unsloth/llama-3-8b-bnb-4bit"
10
 
11
  pipeline = transformers.pipeline(
12
  "text-generation",