torch_dtype -> dtype (#3177)

* torch_dtype -> dtype * torch_dtype -> dtype
2025-10-01 13:32:51 +05:30
parent f4376748f3
commit a6bfbe3400
5 changed files with 6 additions and 8 deletions
--- a/tests/e2e/test_quantization.py
+++ b/tests/e2e/test_quantization.py
@@ -39,7 +39,7 @@ def model():
    dummy_model = AutoModelForCausalLM.from_pretrained(
        "Qwen/Qwen2-0.5B",
        device_map="auto",
-        torch_dtype=torch.bfloat16,
+        dtype=torch.bfloat16,
    )
    with torch.device(dummy_model.device):
        dummy_model.model.embed_tokens = torch.nn.Embedding(