From 59d00da1511510a23e02e3348105aa2b24f104ae Mon Sep 17 00:00:00 2001
From: MekkCyber <mekk.cyber@gmail.com>
Date: Mon, 14 Apr 2025 12:23:11 +0000
Subject: [PATCH] small fix

---
 torchao/core/config.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/torchao/core/config.py b/torchao/core/config.py
index fe03ac225b..aa28e3da2c 100644
--- a/torchao/core/config.py
+++ b/torchao/core/config.py
@@ -175,6 +175,7 @@ def config_to_dict(config: AOBaseConfig) -> Dict[str, Any]:
     "torchao.quantization",
     "torchao.sparsity.sparse_api",
     "torchao.prototype.quantization",
+    "torchao.dtypes",
 }