File size: 425 Bytes
967e5ac 6d3d699 967e5ac 6d3d699 967e5ac 6d3d699 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 |
# # convert_model.py
# from optimum.onnxruntime import ORTModelForSequenceClassification
# # Convert and optimize model
# model = ORTModelForSequenceClassification.from_pretrained(
# "Essay-Grader/roberta-ai-detector-20250401_232702",
# export=True,
# provider="CPUExecutionProvider"
# )
# # Save optimized model
# model.save_pretrained(
# "./optimized_model",
# file_name="model_optimized.onnx"
# )
|