Update demo/t2v.py

demo/t2v.py

@@ -11,6 +11,9 @@ import time
 import torchvision
 from huggingface_hub import hf_hub_download
 
+# IMPORT HF_TOKEN
+hf_token = os.environ['HF_TOKEN']
+
 def save_videos(batch_tensors, savedir, filenames, fps=10):
     # b,samples,c,t,h,w
     n_samples = batch_tensors.shape[1]
@@ -107,7 +110,7 @@ class Text2Video():
             local_file = os.path.join(model_folder, filename)
 
             if not os.path.exists(local_file):
-                hf_hub_download(repo_id=REPO_ID, filename=filename, local_dir=model_folder, local_dir_use_symlinks=False)
+                hf_hub_download(repo_id=REPO_ID, filename=filename, local_dir=model_folder, local_dir_use_symlinks=False, token=hf_token)
 
 if __name__ == '__main__':
     t2v = Text2Video()