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 index 708c631a0..4d5d357da 100644 --- 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,