huggingface/text-generation-inference
5
0
Fork 0
mirror of https://github.com/huggingface/text-generation-inference.git synced 2025-09-11 20:34:54 +00:00
Code Issues Packages Projects Releases Wiki Activity

Update launcher/src/main.rs

Browse Source 
Co-authored-by: Daniël de Kok <me@github.danieldk.eu>
This commit is contained in:
Nicolas Patry 2024-05-23 11:43:38 +02:00 committed by GitHub
parent da1b5fa6b7
commit 1c49da0c2d
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 2 additions and 17 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

19
launcher/src/main.rs
View File

@ -35,23 +35,8 @@ struct Config {
impl From<RawConfig> for Config {
fn from(other: RawConfig) -> Self {
if other.max_position_embeddings.is_some() {
Config {
max_position_embeddings: other.max_position_embeddings,
}
} else if other.max_seq_len.is_some() {
Config {
max_position_embeddings: other.max_seq_len,
}
} else if other.n_positions.is_some() {
Config {
max_position_embeddings: other.n_positions,
}
} else {
Config {
max_position_embeddings: None,
}
}
let max_position_embeddings = other.max_position_embeddings.or(other.max_seq_len).or(other.n_positions);
Config { max_position_embeddings }
}
}