fix eval trust_remote_code (#424)

Signed-off-by: n1ck-guo <[email protected]>
intel · Jan 26, 2025 · 5c075a9 · 5c075a9
1 parent db8088b
commit 5c075a9
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/auto_round/script/llm.py b/auto_round/script/llm.py
@@ -567,7 +567,7 @@ def tune(args):
 
 def _eval_init(tasks, model_path, device, disable_trust_remote_code=False):
     device_str, parallelism = set_cuda_visible_devices(device)
-    model_args = f"pretrained={model_path},trust_remote_code={disable_trust_remote_code}"
+    model_args = f"pretrained={model_path},trust_remote_code={not disable_trust_remote_code}"
     if parallelism:
         model_args += ",parallelize=True"
     if isinstance(tasks, str):