daKhosa commited on
Commit
050a9f8
·
1 Parent(s): 41f5b1e

Align LTX Space runtime dependencies

Browse files
Files changed (2) hide show
  1. app.py +23 -8
  2. requirements.txt +6 -4
app.py CHANGED
@@ -6,6 +6,7 @@ import struct
6
  import subprocess
7
  import sys
8
  import tempfile
 
9
  from pathlib import Path
10
 
11
  import gradio as gr
@@ -71,14 +72,28 @@ def install_ltx_runtime():
71
 
72
  install_ltx_runtime()
73
 
74
- from ltx_core.loader import LTXV_LORA_COMFY_RENAMING_MAP, LoraPathStrengthAndSDOps
75
- from ltx_core.loader.primitives import StateDict
76
- from ltx_core.loader.sft_loader import SafetensorsStateDictLoader
77
- from ltx_core.model.video_vae import TilingConfig, get_video_chunks_number
78
- from ltx_core.quantization import QuantizationPolicy
79
- from ltx_pipelines.distilled import DistilledPipeline
80
- from ltx_pipelines.utils.args import ImageConditioningInput
81
- from ltx_pipelines.utils.media_io import encode_video
 
 
 
 
 
 
 
 
 
 
 
 
 
 
82
 
83
 
84
  _SAFETENSORS_DTYPE_MAP = {
 
6
  import subprocess
7
  import sys
8
  import tempfile
9
+ import builtins
10
  from pathlib import Path
11
 
12
  import gradio as gr
 
72
 
73
  install_ltx_runtime()
74
 
75
+
76
+ _real_import = builtins.__import__
77
+
78
+
79
+ def _import_without_xformers(name, globals=None, locals=None, fromlist=(), level=0):
80
+ if name == "xformers" or name.startswith("xformers."):
81
+ raise ImportError("xformers is disabled for this Space runtime")
82
+ return _real_import(name, globals, locals, fromlist, level)
83
+
84
+
85
+ builtins.__import__ = _import_without_xformers
86
+ try:
87
+ from ltx_core.loader import LTXV_LORA_COMFY_RENAMING_MAP, LoraPathStrengthAndSDOps
88
+ from ltx_core.loader.primitives import StateDict
89
+ from ltx_core.loader.sft_loader import SafetensorsStateDictLoader
90
+ from ltx_core.model.video_vae import TilingConfig, get_video_chunks_number
91
+ from ltx_core.quantization import QuantizationPolicy
92
+ from ltx_pipelines.distilled import DistilledPipeline
93
+ from ltx_pipelines.utils.args import ImageConditioningInput
94
+ from ltx_pipelines.utils.media_io import encode_video
95
+ finally:
96
+ builtins.__import__ = _real_import
97
 
98
 
99
  _SAFETENSORS_DTYPE_MAP = {
requirements.txt CHANGED
@@ -2,15 +2,17 @@ gradio>=5.29.0
2
  spaces
3
  huggingface_hub>=0.24.0
4
 
 
 
5
  einops
6
  numpy
7
- transformers>=4.52
8
  safetensors
9
  accelerate
10
- scipy>=1.14
11
  av
12
  tqdm
13
  Pillow
14
- imageio
15
- imageio-ffmpeg
16
  sentencepiece
 
 
 
2
  spaces
3
  huggingface_hub>=0.24.0
4
 
5
+ torch==2.8.0
6
+ torchaudio==2.8.0
7
  einops
8
  numpy
9
+ transformers==4.57.6
10
  safetensors
11
  accelerate
12
+ scipy
13
  av
14
  tqdm
15
  Pillow
 
 
16
  sentencepiece
17
+ scikit-image>=0.25.2
18
+ flashpack==0.1.2