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

fix: compute type typo (#3150)

Browse Source 
Signed-off-by: Raphael Glon <oOraph@users.noreply.github.com>
Co-authored-by: Raphael Glon <oOraph@users.noreply.github.com>
This commit is contained in:
oOraph 2025-04-07 17:24:11 +02:00 committed by GitHub
parent 37104acd75
commit 24bec29ffc
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 7 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
launcher/src/main.rs
View File

@ -1709,16 +1709,16 @@ impl From<&str> for Gpu {
impl std::fmt::Display for Gpu {
fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
match self {
Gpu::RTX4090 => write!(f, "nvida-4090"),
Gpu::T4 => write!(f, "nvida-t4"),
Gpu::L4 => write!(f, "nvida-l4"),
Gpu::L40 => write!(f, "nvida-l40"),
Gpu::L40S => write!(f, "nvida-l40s"),
Gpu::RTX4090 => write!(f, "nvidia-4090"),
Gpu::T4 => write!(f, "nvidia-t4"),
Gpu::L4 => write!(f, "nvidia-l4"),
Gpu::L40 => write!(f, "nvidia-l40"),
Gpu::L40S => write!(f, "nvidia-l40s"),
Gpu::A10G => write!(f, "nvidia-a10g"),
Gpu::A40 => write!(f, "nvidia-a40"),
Gpu::H100 => write!(f, "nvidia-h100-80fb-hbm3"),
Gpu::A100 => write!(f, "nvida-a100-sxm4-80gb"),
Gpu::H200 => write!(f, "nvida-h200"),
Gpu::A100 => write!(f, "nvidia-a100-sxm4-80gb"),
Gpu::H200 => write!(f, "nvidia-h200"),
Gpu::Unknown(card) => write!(f, "{}", card),
}
}