fix ut

xadupre · xadupre · commit 4ccfcf91cdcc · 2026-01-16T17:15:37.000+01:00
diff --git a/onnx_diagnostic/torch_models/untrained/llm_phi2.py b/onnx_diagnostic/torch_models/untrained/llm_phi2.py
@@ -53,6 +53,7 @@ def get_phi2(
         "transformers_version": "4.37.0",
         "use_cache": True,
         "vocab_size": 51200,
+        "pad_token_id": 0,
     }
     config.update(**kwargs)
     conf = transformers.PhiConfig(**config)

Original file line number	Diff line number	Diff line change
`@@ -53,6 +53,7 @@ def get_phi2(`
`53`	`53`	`"transformers_version": "4.37.0",`
`54`	`54`	`"use_cache": True,`
`55`	`55`	`"vocab_size": 51200,`
	`56`	`+ "pad_token_id": 0,`
`56`	`57`	`}`
`57`	`58`	`config.update(**kwargs)`
`58`	`59`	`conf = transformers.PhiConfig(**config)`