Mirrors
/
hf_text-generation-inference
mirror of https://github.com/huggingface/text-generation-inference.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

fix: adjust types

This commit is contained in:
drbh 2024-06-20 00:56:34 +00:00
parent 75dccec40d
commit abfc581030
1 changed files with 4 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
launcher/src/main.rs
View File

@ -486,7 +486,7 @@ fn shard_manager(
max_batch_size: Option<usize>, max_batch_size: Option<usize>,
max_input_tokens: usize, max_input_tokens: usize,
otlp_endpoint: Option<String>, otlp_endpoint: Option<String>,
otlp_service_name: String, otlp_service_name: Option<String>,
log_level: LevelFilter, log_level: LevelFilter,
status_sender: mpsc::Sender<ShardStatus>, status_sender: mpsc::Sender<ShardStatus>,
shutdown: Arc<AtomicBool>, shutdown: Arc<AtomicBool>,
@ -559,7 +559,7 @@ fn shard_manager(
} }
// OpenTelemetry Service Name // OpenTelemetry Service Name
if let Some(otlp_endpoint) = otlp_endpoint { if let Some(otlp_service_name) = otlp_service_name {
shard_args.push("--otlp-service-name".to_string()); shard_args.push("--otlp-service-name".to_string());
shard_args.push(otlp_service_name); shard_args.push(otlp_service_name);
} }
@ -1220,8 +1220,9 @@ fn spawn_webserver(
} }
// OpenTelemetry // OpenTelemetry
if args.otlp_service_name { if let Some(otlp_service_name) = args.otlp_service_name {
router_args.push("--otlp-service-name".to_string()); router_args.push("--otlp-service-name".to_string());
router_args.push(otlp_service_name);
} }
// CORS origins // CORS origins