set python intepreter, fix path

This commit is contained in:
Brandon Thomas committed 2023-04-04 03:14:40 -07:00
1 parent b844191db7
commit 85abbd8587
3 files changed
+6 -2

No files matched your search

@@ -31,4 +31,5 @@ HIFIGAN_MODEL_CHECK_MAYBE_DOCKER_IMAGE=5642d0fd7fc1
VITS_MODEL_CHECK_ROOT_DIRECTORY="/vits"
VITS_MODEL_CHECK_COMMAND="export_ts.py"
#VITS_MODEL_CHECK_MAYBE_VENV_COMMAND="source python/bin/activate"
VITS_MODEL_CHECK_MAYBE_PYTHON_INTERPRETER="python3"
VITS_MODEL_CHECK_MAYBE_DOCKER_IMAGE="1d5b8788ac28"
@@ -37,7 +37,7 @@ pub async fn process_vits_model<'a, 'b>(
let original_model_file_path = PathBuf::from(download_filename.clone());
// NB: We'll be creating the traced model in the "check" step and uploading it to GDrive along with the original.
let traced_model_file_path = PathBuf::from(format!("{:?}_traced", original_model_file_path));
let traced_model_file_path = PathBuf::from(format!("{:?}_traced", original_model_file_path.as_os_str()));
let config_path = PathBuf::from("configs/ljs_li44_tmbert_nmp_s1_arpa.json"); // TODO: This could be variable.
+4 -1
View File
@@ -266,6 +266,9 @@ async fn main() -> AnyhowResult<()> {
let maybe_venv_command = easyenv::get_env_string_optional(
"VITS_MODEL_CHECK_MAYBE_VENV_COMMAND");
let maybe_python_interpreter = easyenv::get_env_string_optional(
"VITS_MODEL_CHECK_MAYBE_PYTHON_INTERPRETER");
let maybe_docker_options = easyenv::get_env_string_optional(
"VITS_MODEL_CHECK_MAYBE_DOCKER_IMAGE")
.map(|image_name| {
@@ -279,7 +282,7 @@ async fn main() -> AnyhowResult<()> {
VitsModelCheckCommand::new(
root_directory,
check_script,
None,
maybe_python_interpreter.as_deref(),
maybe_venv_command.as_deref(),
maybe_docker_options,
)