svjack commited on
Commit
61e72f4
·
verified ·
1 Parent(s): eea4144

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -3
app.py CHANGED
@@ -25,8 +25,8 @@ os.putenv("HF_HUB_ENABLE_HF_TRANSFER","1")
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"
29
- single_file_url = "https://huggingface.co/Lightricks/LTX-Video/ltx-video-2b-v0.9.5.safetensors"
30
 
31
  pipe = LTXImageToVideoPipeline.from_pretrained(
32
  "Lightricks/LTX-Video",
@@ -35,7 +35,9 @@ pipe = LTXImageToVideoPipeline.from_pretrained(
35
  text_encoder=None,
36
  ).to(torch.device("cuda"),torch.bfloat16)
37
 
38
- text_encoder = T5EncoderModel.from_pretrained("Lightricks/LTX-Video",subfolder='text_encoder',token=True).to(torch.device("cuda"),torch.bfloat16)
 
 
39
  transformer = LTXVideoTransformer3DModel.from_single_file(single_file_url,
40
  #token=HF_TOKEN
41
  ).to(torch.device("cuda"),torch.bfloat16)
 
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"
29
+ #single_file_url = "https://huggingface.co/Lightricks/LTX-Video/ltx-video-2b-v0.9.5.safetensors"
30
 
31
  pipe = LTXImageToVideoPipeline.from_pretrained(
32
  "Lightricks/LTX-Video",
 
35
  text_encoder=None,
36
  ).to(torch.device("cuda"),torch.bfloat16)
37
 
38
+ text_encoder = T5EncoderModel.from_pretrained("Lightricks/LTX-Video",subfolder='text_encoder',
39
+ #token=True
40
+ ).to(torch.device("cuda"),torch.bfloat16)
41
  transformer = LTXVideoTransformer3DModel.from_single_file(single_file_url,
42
  #token=HF_TOKEN
43
  ).to(torch.device("cuda"),torch.bfloat16)