enhance code

Signed-off-by: yuwenzho <[email protected]>
onnx · Jul 24, 2024 · f1440e9 · f1440e9
1 parent efed353
commit f1440e9
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/examples/nlp/huggingface_model/text_generation/quantization/weight_only/prepare_model.py b/examples/nlp/huggingface_model/text_generation/quantization/weight_only/prepare_model.py
@@ -40,6 +40,7 @@ def prepare_model(input_model, output_model, task):
             "--task",
             task,
             f"{output_model}",
+            "--trust-remote-code",
         ],
         stdout=subprocess.PIPE,
         text=True,