From 3b07d6a6f729193a9ed920a2a950c38311534c2f Mon Sep 17 00:00:00 2001 From: Arnav Garg <106701836+arnavgarg1@users.noreply.github.com> Date: Wed, 17 Apr 2024 21:59:38 -0700 Subject: [PATCH] small typo in dequantization script (#3993) --- examples/llm_base_model_dequantization/phi_2_dequantization.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/examples/llm_base_model_dequantization/phi_2_dequantization.py b/examples/llm_base_model_dequantization/phi_2_dequantization.py index 7b818cf344b..b406c466b9e 100644 --- a/examples/llm_base_model_dequantization/phi_2_dequantization.py +++ b/examples/llm_base_model_dequantization/phi_2_dequantization.py @@ -1,4 +1,5 @@ import logging +import os import yaml from huggingface_hub import whoami @@ -10,7 +11,7 @@ base_model_name = "microsoft/phi-2" dequantized_path = "microsoft-phi-2-dequantized" save_path = "/home/ray/" + dequantized_path -hfhub_repo_id = hf_username + dequantized_path +hfhub_repo_id = os.path.join(hf_username, dequantized_path) config = yaml.safe_load(