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

fix: compute type typo

Browse Source 
Signed-off-by: Raphael Glon <oOraph@users.noreply.github.com>
This commit is contained in:
Raphael Glon 2025-04-07 17:10:03 +02:00
parent 37104acd75
commit 3ff8a1d130
No known key found for this signature in database
GPG Key ID: 4D4CC6881E12A0C3
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),
}
}