Update modeleval.py
Browse files- 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 |
-
)
|
| 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)
|