Spaces:
Running
on
Zero
Running
on
Zero
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)
|