sreyanghosh commited on
Commit
007fe97
·
1 Parent(s): 4705650

update/removed quantization

Browse files
Files changed (2) hide show
  1. app.py +1 -1
  2. requirements.txt +1 -1
app.py CHANGED
@@ -17,7 +17,7 @@ def load_model():
17
 
18
  model = AutoPeftModelForCausalLM.from_pretrained(
19
  lora_model_name, # YOUR MODEL YOU USED FOR TRAINING
20
- load_in_4bit = True, # False
21
  )
22
  tokenizer = AutoTokenizer.from_pretrained(lora_model_name)
23
  model.eval()
 
17
 
18
  model = AutoPeftModelForCausalLM.from_pretrained(
19
  lora_model_name, # YOUR MODEL YOU USED FOR TRAINING
20
+ load_in_4bit = False, # False
21
  )
22
  tokenizer = AutoTokenizer.from_pretrained(lora_model_name)
23
  model.eval()
requirements.txt CHANGED
@@ -3,4 +3,4 @@ gradio
3
  transformers
4
  peft
5
  torch
6
- bitsandbytes
 
3
  transformers
4
  peft
5
  torch
6
+ # bitsandbytes