YangZhoumill commited on
Commit
8b4d7b7
1 Parent(s): 6a771c3

pre-release

Browse files
Files changed (1) hide show
  1. model/model_manager.py +2 -0
model/model_manager.py CHANGED
@@ -119,6 +119,8 @@ class ModelManager:
119
  root_dir = SSH_CACHE_SORA
120
 
121
  local_dir = "./cache_video"
 
 
122
  prompt, results = get_ssh_random_video_prompt(root_dir, local_dir, model_names)
123
  cache_dir = local_dir
124
  # cache_dir, prompt = get_random_video_prompt(root_dir)
 
119
  root_dir = SSH_CACHE_SORA
120
 
121
  local_dir = "./cache_video"
122
+ if not os.path.exists(local_dir):
123
+ os.makedirs(local_dir)
124
  prompt, results = get_ssh_random_video_prompt(root_dir, local_dir, model_names)
125
  cache_dir = local_dir
126
  # cache_dir, prompt = get_random_video_prompt(root_dir)