deepjavalibrary / djl-serving

A universal scalable machine learning model deployment solution
Apache License 2.0
182 stars 59 forks source link

[CI] Add trust remote code option in TRTLLM CI #2134

Closed ydm-amazon closed 6 days ago

ydm-amazon commented 6 days ago

Fixes the following CI error:

ValueError: The repository for ccdv/cnn_dailymail contains custom code which must be executed to correctly load the dataset. You can inspect the repository content at https://hf.co/datasets/ccdv/cnn_dailymail.
sindhuvahinis commented 6 days ago

No this is not it. This error is because of dataset version in HuggingFace. We dont need trust_remote_code

ydm-amazon commented 6 days ago

Closed PR since the root cause seems to be elsewhere