chore: fix lib rename

huggingface · Nov 5, 2024 · e63e212 · e63e212
1 parent 8c6e36d
commit e63e212
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/extra/k8s/inference-benchmarker/templates/benchmark.yaml b/extra/k8s/inference-benchmarker/templates/benchmark.yaml
@@ -59,7 +59,7 @@ spec:
                   name: {{ include "inference-benchmarker.fullname" . }}-hf-token
                   key: HF_TOKEN
             - name: RUST_LOG
-              value: "text_generation_inference_benchmark=info"
+              value: "inference_benchmarker=info"
           resources:
             {{- toYaml .Values.benchmark.resources | nindent 12 }}
           volumeMounts:

diff --git a/src/lib.rs b/src/lib.rs
@@ -98,7 +98,7 @@ pub async fn run(run_config: RunConfiguration, stop_sender: Sender<()>) -> anyho
         env_logger::Builder::new()
             .target(env_logger::Target::Pipe(target))
             .filter(
-                Some("text_generation_inference_benchmark"),
+                Some("inference_benchmarker"),
                 LevelFilter::Debug,
             )
             .format(|buf, record| {

diff --git a/src/main.rs b/src/main.rs
@@ -4,7 +4,7 @@ use log::{debug, error};
 use reqwest::Url;
 use std::collections::HashMap;
 use std::time::Duration;
-use text_generation_inference_benchmark::{run, RunConfiguration, TokenizeOptions};
+use inference_benchmarker::{run, RunConfiguration, TokenizeOptions};
 use tokio::sync::broadcast;
 
 #[derive(Parser, Debug)]