fix uncorrect library import

namgyu-youn · namgyu-youn · commit cba6d5df787c · 2025-08-02T18:58:02.000+09:00
diff --git a/test/test_ao_models.py b/test/test_ao_models.py
@@ -6,9 +6,9 @@
 import unittest
 
 import torch
+from torch.testing._internal import common_utils
 
 from torchao._models.llama.model import Transformer
-from torchao.testing import common_utils
 
 
 def init_model(name="stories15M", device="cpu", precision=torch.bfloat16):
