chore: import upstream snapshot with attribution

2026-07-13 13:23:58 +08:00
commit 770d92cb1f
694 changed files with 114634 additions and 0 deletions
@@ -0,0 +1,21 @@
+import pytest
+
+from mlc_llm.model import MODEL_PRESETS, MODELS
+
+
+@pytest.mark.parametrize("model_name", ["phi-1_5", "phi-2"])
+def test_phi_creation(model_name: str):
+    model_info = MODELS["phi-msft"]
+    config = model_info.config.from_dict(MODEL_PRESETS[model_name])
+    model = model_info.model(config)
+    mod, named_params = model.export_tvm(
+        spec=model.get_default_spec(),
+    )
+    mod.show(black_format=False)
+    for name, param in named_params:
+        print(name, param.shape, param.dtype)
+
+
+if __name__ == "__main__":
+    test_phi_creation("phi-1_5")
+    test_phi_creation("phi-2")