File tree Expand file tree Collapse file tree 4 files changed +761
-0
lines changed
components/models/mistral3
tests/unit_tests/models/mistral3 Expand file tree Collapse file tree 4 files changed +761
-0
lines changed Original file line number Diff line number Diff line change @@ -334,6 +334,7 @@ def _retry(**override):
334334 try :
335335 # if we have a custom model implementation available, we prioritize that over HF
336336 if hf_config .architectures [0 ] in ModelRegistry .model_arch_name_to_cls :
337+ kwargs .pop ("trust_remote_code" , None )
337338 model = ModelRegistry .model_arch_name_to_cls [hf_config .architectures [0 ]](
338339 hf_config , * model_args , ** kwargs
339340 )
@@ -508,6 +509,7 @@ def _retry(**override):
508509 try :
509510 # if we have a custom model implementation available, we prioritize that over HF
510511 if config .architectures [0 ] in ModelRegistry .model_arch_name_to_cls :
512+ kwargs .pop ("trust_remote_code" , None )
511513 model = ModelRegistry .model_arch_name_to_cls [config .architectures [0 ]](
512514 config , * model_args , ** kwargs
513515 )
Original file line number Diff line number Diff line change 1+ # Copyright (c) 2025, NVIDIA CORPORATION. All rights reserved.
2+ #
3+ # Licensed under the Apache License, Version 2.0 (the "License");
4+ # you may not use this file except in compliance with the License.
5+ # You may obtain a copy of the License at
6+ #
7+ # http://www.apache.org/licenses/LICENSE-2.0
8+ #
9+ # Unless required by applicable law or agreed to in writing, software
10+ # distributed under the License is distributed on an "AS IS" BASIS,
11+ # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12+ # See the License for the specific language governing permissions and
13+ # limitations under the License.
You can’t perform that action at this time.
0 commit comments