Skip to content

Commit cfb9210

Browse files
authored
Add support for Q8 in DType (#103)
1 parent 86533d9 commit cfb9210

2 files changed

Lines changed: 3 additions & 24 deletions

File tree

src/models/yolo/README.md

Lines changed: 0 additions & 24 deletions
Original file line numberDiff line numberDiff line change
@@ -15,27 +15,3 @@ The official repository can be found on:
1515
## Example
1616

1717
Refer to the [example](../../../examples/yolo)
18-
19-
20-
## TODO
21-
- [x] YOLOv5-det
22-
- [x] YOLOv5-cls
23-
- [x] YOLOv5-seg
24-
- [x] YOLOv6
25-
- [x] YOLOv7
26-
- [x] YOLOv8-det
27-
- [x] YOLOv8-cls
28-
- [x] YOLOv8-pose
29-
- [x] YOLOv8-seg
30-
- [x] YOLOv8-obb
31-
- [x] YOLOv8-world
32-
- [x] YOLOv8-rtdetr
33-
- [x] YOLOv9
34-
- [x] YOLOv10
35-
- [x] YOLO11-det
36-
- [x] YOLO11-cls
37-
- [x] YOLO11-pose
38-
- [x] YOLO11-seg
39-
- [x] YOLO11-obb
40-
- [x] FastSam
41-
- [ ] YOLO-NAS

src/utils/dtype.rs

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -20,6 +20,7 @@ pub enum DType {
2020
Bnb4,
2121
Q4,
2222
Q4f16,
23+
Q8,
2324
}
2425

2526
impl TryFrom<&str> for DType {
@@ -43,6 +44,7 @@ impl TryFrom<&str> for DType {
4344
"b16" | "bf16" => Ok(Self::Bf16),
4445
"q4f16" => Ok(Self::Q4f16),
4546
"q4" => Ok(Self::Q4),
47+
"q8" => Ok(Self::Q8),
4648
"bnb4" => Ok(Self::Bnb4),
4749
x => anyhow::bail!("Unsupported DType: {}", x),
4850
}
@@ -71,6 +73,7 @@ impl std::fmt::Display for DType {
7173
Self::Bnb4 => "bnb4",
7274
Self::Q4 => "q4",
7375
Self::Q4f16 => "q4f16",
76+
Self::Q8 => "q8",
7477
};
7578
write!(f, "{}", x)
7679
}

0 commit comments

Comments
 (0)