Skip to content

Commit

Permalink
small typo in dequantization script (#3993)
Browse files Browse the repository at this point in the history
  • Loading branch information
arnavgarg1 authored Apr 18, 2024
1 parent db4e371 commit 3b07d6a
Showing 1 changed file with 2 additions and 1 deletion.
Original file line number Diff line number Diff line change
@@ -1,4 +1,5 @@
import logging
import os

import yaml
from huggingface_hub import whoami
Expand All @@ -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(
Expand Down

0 comments on commit 3b07d6a

Please sign in to comment.