Add buck target for hf_download (pytorch#9603)

Summary: Pull Request resolved: pytorch#9603

Reviewed By: kirklandsign

Differential Revision: D71833608

Author: jackzhxng

examples/models/llama/TARGETS

@@ -118,6 +118,16 @@ runtime.python_library(
     ],
 )
 
+runtime.python_library(
+    name = "hf_download",
+    srcs = [
+        "hf_download.py",
+    ],
+    deps = [
+        "fbsource//third-party/pypi/huggingface-hub:huggingface-hub",
+    ]
+)
+
 runtime.python_library(
     name = "export_library",
     srcs = [
@@ -134,6 +144,7 @@ runtime.python_library(
         "@EXECUTORCH_CLIENTS",
     ],
     deps = [
+        ":hf_download",
         ":source_transformation",
         "//ai_codesign/gen_ai/fast_hadamard_transform:fast_hadamard_transform",
         "//caffe2:torch",

examples/models/llama/export_llama_lib.py

@@ -539,16 +539,24 @@ def export_llama(args) -> str:
     if not args.checkpoint and args.model in HUGGING_FACE_REPO_IDS:
         repo_id = HUGGING_FACE_REPO_IDS[args.model]
         if args.model == "qwen2_5":
-            from executorch.examples.models.qwen2_5 import convert_weights
+            from executorch.examples.models.qwen2_5 import (  # pyre-ignore[21]
+                convert_weights,
+            )
         elif args.model == "phi_4_mini":
-            from executorch.examples.models.phi_4_mini import convert_weights
+            from executorch.examples.models.phi_4_mini import (  # pyre-ignore[21]
+                convert_weights,
+            )
         elif args.model == "smollm2":
-            from executorch.examples.models.smollm2 import convert_weights
+            from executorch.examples.models.smollm2 import (  # pyre-ignore[21]
+                convert_weights,
+            )
         else:
             raise ValueError(
                 f"Converting weights to meta format for {args.model} is not yet supported"
             )
-        args.checkpoint = download_and_convert_hf_checkpoint(repo_id, convert_weights)
+        args.checkpoint = download_and_convert_hf_checkpoint(
+            repo_id, convert_weights
+        )  # pyre-ignore
 
     if args.profile_path is not None:
         try:

examples/models/llama/hf_download.py

@@ -33,7 +33,7 @@ def download_and_convert_hf_checkpoint(
 
     # Use repo name to name the converted file.
     model_name = repo_id.replace("/", "_")
-    converted_path = cache_dir / f"{model_name}.pth"
+    converted_path = str(cache_dir / f"{model_name}.pth")
 
     if converted_path.exists():
         print(f"✔ Using cached converted model: {converted_path}")