Fix preq embedding dtype check (#10699)

Summary:



Rollback Plan:

Differential Revision: D74194622

Author: YIWENX14

examples/models/llama/source_transformation/pre_quantization.py

@@ -146,7 +146,7 @@ def filter_fn(child: torch.nn.Module, cur_fqn: str) -> bool:
         scales_key = f"{cur_fqn}.scales"
         if isinstance(child, nn.Embedding) and scales_key in checkpoint:
             assert checkpoint[f"{cur_fqn}.weight"].dtype == torch.int8
-            assert checkpoint[scales_key].dtype == torch.float32
+            assert checkpoint[scales_key].dtype == dtype
             return True
         return False