feat(autogpt/transformers): consume trust_remote_code (#1799)

trusting remote code by default is a danger to our users
2025-06-12 20:18:11 +00:00 · 2024-03-05 13:47:15 -05:00
parent 504f2e8bf4
commit 5c69dd155f
2 changed files with 3 additions and 3 deletions
--- a/backend/python/autogptq/autogptq.py
+++ b/backend/python/autogptq/autogptq.py
@ -33,7 +33,7 @@ class BackendServicer(backend_pb2_grpc.BackendServicer):
            model = AutoGPTQForCausalLM.from_quantized(request.Model,
                    model_basename=request.ModelBaseName,
                    use_safetensors=True,
-                    trust_remote_code=True,
+                    trust_remote_code=request.TrustRemoteCode,
                    device=device,
                    use_triton=request.UseTriton,
                    quantize_config=None)