Skip to content

Commit 934b237

Browse files
Update format-specific tags in ModelCardGenerator: add 'transformers' tag to GGUF, ONNX, and MLX formats for improved compatibility and integration.
1 parent fa63aec commit 934b237

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

quantllm/hub/model_card.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -31,9 +31,9 @@ class ModelCardGenerator:
3131

3232
# Format-specific tags
3333
FORMAT_TAGS = {
34-
"gguf": ["gguf", "llama-cpp", "quantized"],
35-
"onnx": ["onnx", "onnxruntime"],
36-
"mlx": ["mlx", "mlx-lm", "apple-silicon"],
34+
"gguf": ["gguf", "llama-cpp", "quantized", "transformers"],
35+
"onnx": ["onnx", "onnxruntime", "transformers"],
36+
"mlx": ["mlx", "mlx-lm", "apple-silicon", "transformers"],
3737
"safetensors": ["transformers", "safetensors"],
3838
"pytorch": ["transformers", "pytorch"],
3939
}

0 commit comments

Comments
 (0)