deepnet commited on
Commit
072f790
·
verified ·
1 Parent(s): 95a751f

Update modeleval.py

Browse files
Files changed (1) hide show
  1. modeleval.py +2 -2
modeleval.py CHANGED
@@ -128,8 +128,8 @@ def evaluate_single_model(model_path, tokenizer_path, csv_path, seq_length, batc
128
  model_path,
129
  # quantization_config=bnb_config, # Use quantization
130
  torch_dtype=torch.float16, # 4-bit models compute in FP32
131
- device_map="auto"
132
- )#.to(device)
133
 
134
  # Convert model to bfloat16
135
  # model.to(torch.bfloat16)
 
128
  model_path,
129
  # quantization_config=bnb_config, # Use quantization
130
  torch_dtype=torch.float16, # 4-bit models compute in FP32
131
+ # device_map="auto"
132
+ ).to(device)
133
 
134
  # Convert model to bfloat16
135
  # model.to(torch.bfloat16)