GIGAParviz commited on
Commit
1e61d2d
·
verified ·
1 Parent(s): 99fa0bf

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -5,7 +5,7 @@ import torch
5
  from transformers import TextStreamer
6
 
7
  model, tokenizer = FastLanguageModel.from_pretrained(
8
- model_name = "nsloth/llama-2-7b-bnb-4bit",
9
  max_seq_length = 2048,
10
  dtype = torch.float16,
11
  load_in_4bit = True,
 
5
  from transformers import TextStreamer
6
 
7
  model, tokenizer = FastLanguageModel.from_pretrained(
8
+ model_name = "unsloth/llama-2-7b-bnb-4bit",
9
  max_seq_length = 2048,
10
  dtype = torch.float16,
11
  load_in_4bit = True,