1inkusFace commited on
Commit
a96a646
·
verified ·
1 Parent(s): 717cdc1

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -22,7 +22,7 @@ torch.backends.cudnn.benchmark = False
22
  torch.set_float32_matmul_precision("highest")
23
  os.putenv("HF_HUB_ENABLE_HF_TRANSFER","1")
24
  HF_TOKEN = os.getenv("HF_TOKEN")
25
-
26
  MAX_SEED = np.iinfo(np.int64).max
27
 
28
  single_file_url = "https://huggingface.co/Lightricks/LTX-Video/ltx-video-2b-v0.9.1.safetensors"
 
22
  torch.set_float32_matmul_precision("highest")
23
  os.putenv("HF_HUB_ENABLE_HF_TRANSFER","1")
24
  HF_TOKEN = os.getenv("HF_TOKEN")
25
+ os.environ["SAFETENSORS_FAST_GPU"] = "1"
26
  MAX_SEED = np.iinfo(np.int64).max
27
 
28
  single_file_url = "https://huggingface.co/Lightricks/LTX-Video/ltx-video-2b-v0.9.1.safetensors"