abdfajar707 commited on
Commit
9fc7678
·
verified ·
1 Parent(s): 9de2232

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -1
app.py CHANGED
@@ -42,7 +42,10 @@ model = AutoPeftModelForCausalLM.from_pretrained(
42
  model_id, # YOUR MODEL YOU USED FOR TRAINING
43
  load_in_4bit = load_in_4bit,
44
  )
45
- tokenizer = AutoTokenizer.from_pretrained(model_id)
 
 
 
46
 
47
  model.config.sliding_window = 4096
48
  model.eval()
 
42
  model_id, # YOUR MODEL YOU USED FOR TRAINING
43
  load_in_4bit = load_in_4bit,
44
  )
45
+ tokenizer = AutoTokenizer.from_pretrained(
46
+ model_id,
47
+ quantization_config=BitsAndBytesConfig(load_in_8bit=True)
48
+ )
49
 
50
  model.config.sliding_window = 4096
51
  model.eval()