Spaces:
Running
on
Zero
Running
on
Zero
YangZhoumill
commited on
Commit
•
8b4d7b7
1
Parent(s):
6a771c3
pre-release
Browse files- 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)
|