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: wrap python basic logs in debug assertion in launcher

Browse Source
This commit is contained in:
OlivierDehaene 2024-09-20 14:31:55 +02:00
parent f478aa77ad
commit 6c9b5de2b7
No known key found for this signature in database
GPG Key ID: BB104D67809DA93C
1 changed files with 7 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
launcher/src/main.rs
View File

@ -1038,6 +1038,8 @@ fn log_lines<R: Sized + Read>(mut bufread: BufReader<R>) {
Ok(log) => log.trace(), Ok(log) => log.trace(),
// For interactive debugging ? // For interactive debugging ?
Err(_) => { Err(_) => {
#[cfg(debug_assertions)]
{
stdout.write_all(line).unwrap(); stdout.write_all(line).unwrap();
if lines.peek().is_some() { if lines.peek().is_some() {
stdout.write_all(b"\n").unwrap(); stdout.write_all(b"\n").unwrap();
@ -1049,6 +1051,7 @@ fn log_lines<R: Sized + Read>(mut bufread: BufReader<R>) {
} }
} }
} }
}
} }
fn find_num_shards( fn find_num_shards(