Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
|
@@ -6,6 +6,10 @@ import subprocess
|
|
| 6 |
import uuid
|
| 7 |
import shutil
|
| 8 |
from tqdm import tqdm
|
|
|
|
|
|
|
|
|
|
|
|
|
| 9 |
|
| 10 |
|
| 11 |
from huggingface_hub import snapshot_download, list_repo_files, hf_hub_download
|
|
@@ -291,6 +295,7 @@ def run_pipeline(prompt_text, steps, image_paths, audio_file_path, max_duration
|
|
| 291 |
tea_cache_l1_thresh = 0.0,
|
| 292 |
progress_bar_cmd=progress
|
| 293 |
)
|
|
|
|
| 294 |
|
| 295 |
# Return resulting video path
|
| 296 |
video_path = os.path.join(output_dir, f"{filename}.mp4")
|
|
|
|
| 6 |
import uuid
|
| 7 |
import shutil
|
| 8 |
from tqdm import tqdm
|
| 9 |
+
import threading
|
| 10 |
+
|
| 11 |
+
def print_thread_count(tag=""):
|
| 12 |
+
print(f"[DEBUG] {tag} Thread count:", len(threading.enumerate()))
|
| 13 |
|
| 14 |
|
| 15 |
from huggingface_hub import snapshot_download, list_repo_files, hf_hub_download
|
|
|
|
| 295 |
tea_cache_l1_thresh = 0.0,
|
| 296 |
progress_bar_cmd=progress
|
| 297 |
)
|
| 298 |
+
print_thread_count("after generation")
|
| 299 |
|
| 300 |
# Return resulting video path
|
| 301 |
video_path = os.path.join(output_dir, f"{filename}.mp4")
|