Andreas Varvarigos commited on
Commit
c71f098
·
verified ·
1 Parent(s): 8efda69

Update src/litbench_pipeline.py

Browse files
Files changed (1) hide show
  1. src/litbench_pipeline.py +1 -7
src/litbench_pipeline.py CHANGED
@@ -623,14 +623,8 @@ if __name__ == "__main__":
623
 
624
 
625
  # Load model and tokenizer
626
- bnb_config = BitsAndBytesConfig(
627
- load_in_8bit=True,
628
- bnb_8bit_use_double_quant=True,
629
- bnb_8bit_quant_type="nf8",
630
- bnb_8bit_compute_dtype=torch.bfloat16
631
- )
632
  tokenizer = AutoTokenizer.from_pretrained(model_name)
633
- model = AutoModelForCausalLM.from_pretrained(model_name, quantization_config=bnb_config)
634
  if model.device.type != 'cuda':
635
  model.to('cuda')
636
 
 
623
 
624
 
625
  # Load model and tokenizer
 
 
 
 
 
 
626
  tokenizer = AutoTokenizer.from_pretrained(model_name)
627
+ model = AutoModelForCausalLM.from_pretrained(model_name)
628
  if model.device.type != 'cuda':
629
  model.to('cuda')
630