[QEff. Finetune] : Use login_and_download_hf_lm in finetuning path (q…

…uic#232)

Use login_and_download_hf_lm function in finetuning path

---------

Signed-off-by: Mamta Singh <[email protected]>

QEfficient/cloud/finetune.py

@@ -30,6 +30,7 @@
     get_preprocessed_dataset,
 )
 from QEfficient.finetune.utils.train_utils import get_longest_seq_length, print_model_size, train
+from QEfficient.utils._utils import login_and_download_hf_lm
 
 try:
     import torch_qaic  # noqa: F401
@@ -76,8 +77,9 @@ def main(**kwargs):
 
     # Load the pre-trained model and setup its configuration
     # config = AutoConfig.from_pretrained(train_config.model_name)
+    pretrained_model_path = login_and_download_hf_lm(train_config.model_name)
     model = AutoModelForCausalLM.from_pretrained(
-        train_config.model_name,
+        pretrained_model_path,
         use_cache=False,
         attn_implementation="sdpa",
         torch_dtype=torch.float16,