huggingface/text-generation-inference
5
0
Fork 0
mirror of https://github.com/huggingface/text-generation-inference.git synced 2025-09-12 04:44:52 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix errors

Browse Source
This commit is contained in:
erikkaum 2024-07-16 16:03:41 +02:00
parent 47a4f1fd00
commit cfc1187048
2 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
router/src/main.rs
View File

@ -382,8 +382,8 @@ async fn main() -> Result<(), RouterError> {
}
};
// Only send usage stats when TGI is run in container
let is_container = usage_stats::is_container();
// Only send usage stats when TGI is run in container and the function returns Some
let is_container = matches!(usage_stats::is_container(), Ok(true));
let user_agent = if !disable_usage_stats && is_container {
let reduced_args = usage_stats::Args::new(

2
router/src/usage_stats.rs
View File

@ -1,7 +1,7 @@
use crate::config::Config;
use reqwest::header::HeaderMap;
use serde::Serialize;
use std::{fmt, process::Command, time::Duration};
use std::{fs::File, io::{self, BufRead}, path::Path, process::Command, time::Duration};
use uuid::Uuid;
use csv::ReaderBuilder;