diff --git a/crates/service/job/inference_job/src/job/job_types/studio_gen2/stable_animator_command.rs b/crates/service/job/inference_job/src/job/job_types/studio_gen2/stable_animator_command.rs index 5a570aed11..fdfcf4e9c1 100644 --- a/crates/service/job/inference_job/src/job/job_types/studio_gen2/stable_animator_command.rs +++ b/crates/service/job/inference_job/src/job/job_types/studio_gen2/stable_animator_command.rs @@ -24,17 +24,17 @@ python inference_advanced.py \ --gradient_checkpointing */ +use crate::util::get_filtered_env_vars::get_filtered_env_vars_hashmap; +use errors::AnyhowResult; +use filesys::path_to_string::path_to_string; +use log::{debug, info, warn}; use std::path::{Path, PathBuf}; use std::process::Stdio; use std::time::Duration; -use log::{debug, info, warn}; -use tokio::io::{AsyncBufReadExt, AsyncWriteExt, BufReader}; -use tokio::process::Command; -use errors::AnyhowResult; -use filesys::path_to_string::path_to_string; use subprocess_common::command_exit_status::CommandExitStatus; use subprocess_common::docker_options::DockerOptions; -use crate::util::get_filtered_env_vars::get_filtered_env_vars_hashmap; +use tokio::io::{AsyncBufReadExt, AsyncWriteExt, BufReader}; +use tokio::process::Command; #[derive(Clone)] pub struct StableAnimatorCommand { @@ -88,9 +88,9 @@ pub struct InferenceArgs<'s> { impl StableAnimatorCommand { pub fn new_from_env() -> AnyhowResult { Ok(Self { - root_code_directory: PathBuf::from("/home/bt/dev/storyteller/storyteller-ml/animation/StableAnimator"), + root_code_directory: PathBuf::from("/model_code"), executable_or_command: ExecutableOrCommand::Command("python inference_advanced.py".to_string()), - maybe_virtual_env_activation_command: Some(String::from("source venv/bin/activate")), + maybe_virtual_env_activation_command: Some(String::from("source /python_install/python/bin/activate")), maybe_docker_options: None, maybe_execution_timeout: None, })