eranlevinlt commited on
Commit
0798457
·
1 Parent(s): 6cb251b
Files changed (2) hide show
  1. app.py +2 -2
  2. requirements.txt +1 -1
app.py CHANGED
@@ -1,6 +1,6 @@
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,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
- snapshot_download("Lightricks/Xora", 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 hf_hub_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'
requirements.txt CHANGED
@@ -1,4 +1,4 @@
1
- huggingface_hub==0.25
2
  torch
3
  diffusers==0.28.2
4
  transformers==4.44.2
 
1
+ huggingface_hub
2
  torch
3
  diffusers==0.28.2
4
  transformers==4.44.2