diff --git a/src/schnetpack/utils/__init__.py b/src/schnetpack/utils/__init__.py index 7c9bfe70d..2c85d6015 100644 --- a/src/schnetpack/utils/__init__.py +++ b/src/schnetpack/utils/__init__.py @@ -27,8 +27,6 @@ "qint8": torch.qint8, "qint32": torch.qint32, "bool": torch.bool, - "quint4x2": torch.quint4x2, - "quint2x4": torch.quint2x4, } TORCH_DTYPES.update({"torch." + k: v for k, v in TORCH_DTYPES.items()})