test updates

Signed-off-by: Brian Dellabetta <bdellabe@redhat.com>

Author: brian-dellabetta

src/llmcompressor/modifiers/awq/base.py

@@ -767,7 +767,9 @@ def get_lowest_common_parent(names: List[str], module: Module) -> Tuple[str, Mod
             parent_name = s1[:i].rstrip(".")
             break
 
-    while parent_name != "":
+    while True:
+        if parent_name == "":
+            return "", module
         parent = get_layer_by_name(parent_name, module)
         if not isinstance(parent, torch.nn.ModuleList):
             return parent_name, parent

tests/llmcompressor/modifiers/awq/test_base.py

@@ -202,12 +202,13 @@ def test_get_lowest_common_parent():
     )
     model = torch.nn.ModuleDict(
         {
+            "embed_tokens": torch.nn.Linear(4, 2),
             "decoder": torch.nn.ModuleDict(
                 {
                     "self_attn": self_attn,
                     "mlp": mlp,
                 }
-            )
+            ),
         }
     )
 
@@ -225,3 +226,8 @@ def test_get_lowest_common_parent():
         ["decoder.mlp.experts.1.gate_proj", "decoder.self_attn.v_proj"], model
     )
     assert parent_name == "decoder" and parent == model["decoder"]
+
+    parent_name, parent = get_lowest_common_parent(
+        ["embed_tokens", "decoder.self_attn.v_proj"], model
+    )
+    assert parent_name == "" and parent == model