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: add name arg to tests

This commit is contained in:
drbh 2024-02-15 12:29:22 -05:00 committed by GitHub
parent 9f67f4226b
commit 1135de23d1
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 17 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
router/src/infer.rs
View File

@ -800,18 +800,22 @@ mod tests {
Message { Message {
role: "user".to_string(), role: "user".to_string(),
content: "Hi!".to_string(), content: "Hi!".to_string(),
name: None,
}, },
Message { Message {
role: "assistant".to_string(), role: "assistant".to_string(),
content: "Hello how can I help?".to_string(), content: "Hello how can I help?".to_string(),
name: None,
}, },
Message { Message {
role: "user".to_string(), role: "user".to_string(),
content: "What is Deep Learning?".to_string(), content: "What is Deep Learning?".to_string(),
name: None,
}, },
Message { Message {
role: "assistant".to_string(), role: "assistant".to_string(),
content: "magic!".to_string(), content: "magic!".to_string(),
name: None,
}, },
], ],
bos_token: Some("[BOS]"), bos_token: Some("[BOS]"),
@ -861,22 +865,27 @@ mod tests {
Message { Message {
role: "user".to_string(), role: "user".to_string(),
content: "Hi!".to_string(), content: "Hi!".to_string(),
name: None,
}, },
Message { Message {
role: "user".to_string(), role: "user".to_string(),
content: "Hi again!".to_string(), content: "Hi again!".to_string(),
name: None,
}, },
Message { Message {
role: "assistant".to_string(), role: "assistant".to_string(),
content: "Hello how can I help?".to_string(), content: "Hello how can I help?".to_string(),
name: None,
}, },
Message { Message {
role: "user".to_string(), role: "user".to_string(),
content: "What is Deep Learning?".to_string(), content: "What is Deep Learning?".to_string(),
name: None,
}, },
Message { Message {
role: "assistant".to_string(), role: "assistant".to_string(),
content: "magic!".to_string(), content: "magic!".to_string(),
name: None,
}, },
], ],
bos_token: Some("[BOS]"), bos_token: Some("[BOS]"),
@ -931,18 +940,22 @@ mod tests {
Message { Message {
role: "user".to_string(), role: "user".to_string(),
content: "Hi!".to_string(), content: "Hi!".to_string(),
name: None,
}, },
Message { Message {
role: "assistant".to_string(), role: "assistant".to_string(),
content: "Hello how can I help?".to_string(), content: "Hello how can I help?".to_string(),
name: None,
}, },
Message { Message {
role: "user".to_string(), role: "user".to_string(),
content: "What is Deep Learning?".to_string(), content: "What is Deep Learning?".to_string(),
name: None,
}, },
Message { Message {
role: "assistant".to_string(), role: "assistant".to_string(),
content: "magic!".to_string(), content: "magic!".to_string(),
name: None,
}, },
], ],
bos_token: Some("[BOS]"), bos_token: Some("[BOS]"),
@ -981,18 +994,22 @@ mod tests {
Message { Message {
role: "user".to_string(), role: "user".to_string(),
content: "Hi!".to_string(), content: "Hi!".to_string(),
name: None,
}, },
Message { Message {
role: "assistant".to_string(), role: "assistant".to_string(),
content: "Hello how can I help?".to_string(), content: "Hello how can I help?".to_string(),
name: None,
}, },
Message { Message {
role: "user".to_string(), role: "user".to_string(),
content: "What is Deep Learning?".to_string(), content: "What is Deep Learning?".to_string(),
name: None,
}, },
Message { Message {
role: "assistant".to_string(), role: "assistant".to_string(),
content: "magic!".to_string(), content: "magic!".to_string(),
name: None,
}, },
], ],
bos_token: Some("[BOS]"), bos_token: Some("[BOS]"),