Skip to content

Commit

Permalink
Merge experiment-allow-overwrite
Browse files Browse the repository at this point in the history
  • Loading branch information
thanawan-atc committed Jul 12, 2023
1 parent 1146dc8 commit 32212f3
Showing 1 changed file with 4 additions and 1 deletion.
5 changes: 4 additions & 1 deletion utils/model_uploader/save_model_file_path_to_env.py
Original file line number Diff line number Diff line change
Expand Up @@ -7,10 +7,13 @@

import argparse


def get_model_file_path(model_id, model_version, model_format):
model_name = str(model_id.split("/")[-1])
model_format = model_format.lower()
model_dirname = f"sentence-transformers/upload/{model_name}/{model_version}/{model_format}"
model_dirname = (
f"sentence-transformers/upload/{model_name}/{model_version}/{model_format}"
)
model_filename = (
f"sentence-transformers_{model_name}-{model_version}-{model_format}.zip"
)
Expand Down

0 comments on commit 32212f3

Please sign in to comment.