Tidzo commited on
Commit
76a7207
·
1 Parent(s): 5548617

Update handler.py

Browse files
Files changed (1) hide show
  1. handler.py +0 -1
handler.py CHANGED
@@ -15,7 +15,6 @@ class EndpointHandler():
15
  config = PeftConfig.from_pretrained(path)
16
 
17
  bnb_config = BitsAndBytesConfig(
18
- load_in_4bit=True,
19
  bnb_4bit_quant_type="nf4",
20
  bnb_4bit_compute_dtype=torch.float16,
21
  )
 
15
  config = PeftConfig.from_pretrained(path)
16
 
17
  bnb_config = BitsAndBytesConfig(
 
18
  bnb_4bit_quant_type="nf4",
19
  bnb_4bit_compute_dtype=torch.float16,
20
  )