Update modeling_apex.py
Browse files- modeling_apex.py +4 -1
modeling_apex.py
CHANGED
|
@@ -66,7 +66,10 @@ class APEXModel(PreTrainedModel):
|
|
| 66 |
config.mert_model_name, trust_remote_code=True
|
| 67 |
)
|
| 68 |
self.mert = AutoModel.from_pretrained(
|
| 69 |
-
config.mert_model_name,
|
|
|
|
|
|
|
|
|
|
| 70 |
)
|
| 71 |
self.mert.eval()
|
| 72 |
for param in self.mert.parameters():
|
|
|
|
| 66 |
config.mert_model_name, trust_remote_code=True
|
| 67 |
)
|
| 68 |
self.mert = AutoModel.from_pretrained(
|
| 69 |
+
config.mert_model_name,
|
| 70 |
+
trust_remote_code = True,
|
| 71 |
+
device_map = None,
|
| 72 |
+
low_cpu_mem_usage = False
|
| 73 |
)
|
| 74 |
self.mert.eval()
|
| 75 |
for param in self.mert.parameters():
|