[Hardware][CPU] Update torch 2.5 (#9911)

Signed-off-by: jiang1.li <jiang1.li@intel.com>
2024-11-07 12:43:08 +08:00
parent 29862b884b
commit a4b3e0c1e9
12 changed files with 81 additions and 53 deletions
--- a/tests/models/decoder_only/language/test_models.py
+++ b/tests/models/decoder_only/language/test_models.py
@@ -32,8 +32,7 @@ if not current_platform.is_cpu():
        "openbmb/MiniCPM3-4B",
    ]

-# TODO: remove this after CPU float16 support ready
-target_dtype = "float" if current_platform.is_cpu() else "half"
+target_dtype = "half"


@pytest.mark.parametrize("model", MODELS)