multimodalart HF Staff commited on
Commit
64c422e
·
verified ·
1 Parent(s): f6764d3

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -0
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}'...")