eranlevinlt commited on
Commit
23d8452
·
2 Parent(s): 0798457 db54e16

Merge remote-tracking branch 'origin/main'

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -1,6 +1,6 @@
1
  import gradio as gr
2
  import torch
3
- from huggingface_hub import hf_hub_download
4
 
5
  from xora.models.autoencoders.causal_video_autoencoder import CausalVideoAutoencoder
6
  from xora.models.transformers.transformer3d import Transformer3DModel
@@ -24,7 +24,7 @@ hf_token = os.getenv("HF_TOKEN")
24
  # Set model download directory within Hugging Face Spaces
25
  model_path = "asset"
26
  if not os.path.exists(model_path):
27
- hf_hub_download("Lightricks/LTX-VIDEO", local_dir=model_path, repo_type='model', token=hf_token)
28
 
29
  # Global variables to load components
30
  vae_dir = Path(model_path) / 'vae'
 
1
  import gradio as gr
2
  import torch
3
+ from huggingface_hub import snapshot_download
4
 
5
  from xora.models.autoencoders.causal_video_autoencoder import CausalVideoAutoencoder
6
  from xora.models.transformers.transformer3d import Transformer3DModel
 
24
  # Set model download directory within Hugging Face Spaces
25
  model_path = "asset"
26
  if not os.path.exists(model_path):
27
+ hf_hub_download("Lightricks/LTX-Video", local_dir=model_path, repo_type='model', token=hf_token)
28
 
29
  # Global variables to load components
30
  vae_dir = Path(model_path) / 'vae'