Add text type

huggingface · Feb 14, 2024 · c4bad8c · c4bad8c
1 parent ba8ff36
commit c4bad8c
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/quanto/library/ext/README.md b/quanto/library/ext/README.md
@@ -14,7 +14,7 @@ The operations are defined in `library/ops.py`.
 
 To provide an implementation for specific device types, use the following syntax:
 
-```
+```python
 @torch.library.impl("quanto_ext::unpack", ["CPU", "CUDA"])
 def unpack(packed: torch.Tensor, bits: int) -> torch.Tensor:
     return ext().unpack(t, bits)

diff --git a/quanto/library/python/README.md b/quanto/library/python/README.md
@@ -8,7 +8,7 @@ The operations are defined in `library/ops.py`.
 
 To provide an implementation for an operation, use the following syntax:
 
-```
+```python
 @torch.library.impl("quanto_py::unpack", "default")
 def unpack(packed: torch.Tensor, bits: int) -> torch.Tensor:
     ...