Simplify weight loading in Transformers backend (#21382)

Signed-off-by: Harry Mellor <19981378+hmellor@users.noreply.github.com>
2025-07-23 04:29:43 +01:00
parent 3ec7170ff1
commit f154bb9ff0
7 changed files with 53 additions and 76 deletions
--- a/tests/models/test_transformers.py
+++ b/tests/models/test_transformers.py
@@ -56,7 +56,7 @@ def check_implementation(
    "model,model_impl",
    [
        ("meta-llama/Llama-3.2-1B-Instruct", "transformers"),
-        ("ArthurZ/Ilama-3.2-1B", "auto"),  # CUSTOM CODE
+        ("hmellor/Ilama-3.2-1B", "auto"),  # CUSTOM CODE
    ])  # trust_remote_code=True by default
 def test_models(
    hf_runner: type[HfRunner],