Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
|
@@ -38,6 +38,9 @@ from longcat_video.context_parallel import context_parallel_util
|
|
| 38 |
from transformers import AutoTokenizer, UMT5EncoderModel
|
| 39 |
from diffusers.utils import export_to_video
|
| 40 |
|
|
|
|
|
|
|
|
|
|
| 41 |
# Download model weights from Hugging Face Hub if they don't exist
|
| 42 |
if not os.path.exists(CHECKPOINT_DIR):
|
| 43 |
print(f"Downloading model weights to '{CHECKPOINT_DIR}'...")
|
|
|
|
| 38 |
from transformers import AutoTokenizer, UMT5EncoderModel
|
| 39 |
from diffusers.utils import export_to_video
|
| 40 |
|
| 41 |
+
# Add the cloned repository to the Python path to allow imports
|
| 42 |
+
sys.path.insert(0, os.path.abspath(REPO_PATH))
|
| 43 |
+
|
| 44 |
# Download model weights from Hugging Face Hub if they don't exist
|
| 45 |
if not os.path.exists(CHECKPOINT_DIR):
|
| 46 |
print(f"Downloading model weights to '{CHECKPOINT_DIR}'...")
|