From 6ab109e3c475f93ac3a57ccbf27cd13b2a299493 Mon Sep 17 00:00:00 2001 From: andylolu2 Date: Fri, 25 Mar 2022 03:20:29 +0000 Subject: [PATCH] Fix grammatical mistake --- README.md | 10 +++++++--- 1 file changed, 7 insertions(+), 3 deletions(-) diff --git a/README.md b/README.md index 0b9ea42..c466ece 100644 --- a/README.md +++ b/README.md @@ -28,12 +28,16 @@ Evaluation script can be found in [test_speed.py](tests/test_speed.py). ## :gear: Installation +### Pip + ```terminal pip install speedtoxify ``` +### GPU Inference + Please additionally install `onnxruntime-gpu` for inference on gpus. -Requires the machine have CUDA installed. +Requires the machine to have CUDA installed. ```terminal pip install onnxruntime-gpu @@ -45,9 +49,9 @@ pip install onnxruntime-gpu from speedtoxify import Speedtoxify model = Speedtoxify("original-small") -# Exporting to onnx format to /home/andylo/.cache/detoxify_onnx/original-small.onnx... +# Exporting to onnx format to ~/.cache/detoxify_onnx/original-small.onnx... # Using framework PyTorch: 1.11.0+cu102 -# Removing shared weights from /home/andylo/.cache/detoxify_onnx/original-small.onnx... +# Removing shared weights from ~/.cache/detoxify_onnx/original-small.onnx... # Validating ONNX model... # -[✓] ONNX model output names match reference model ({'logits'}) # - Validating ONNX Model output "logits":