svjack commited on
Commit
0e41673
·
verified ·
1 Parent(s): 47c1220

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +8 -5
app.py CHANGED
@@ -21,21 +21,24 @@ torch.backends.cudnn.benchmark = True
21
  #torch.backends.cuda.preferred_linalg_library="cusolver"
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"
 
29
 
30
  pipe = LTXImageToVideoPipeline.from_pretrained(
31
  "Lightricks/LTX-Video",
32
- token=HF_TOKEN,
33
  transformer=None,
34
  text_encoder=None,
35
  ).to(torch.device("cuda"),torch.bfloat16)
36
 
37
  text_encoder = T5EncoderModel.from_pretrained("Lightricks/LTX-Video",subfolder='text_encoder',token=True).to(torch.device("cuda"),torch.bfloat16)
38
- transformer = LTXVideoTransformer3DModel.from_single_file(single_file_url,token=HF_TOKEN).to(torch.device("cuda"),torch.bfloat16)
 
 
39
 
40
  @spaces.GPU(duration=80)
41
  def generate_video(
@@ -93,4 +96,4 @@ iface = gr.Interface(
93
  description="Generate video from image with LTX-Image-to-Video.",
94
  )
95
 
96
- iface.launch()
 
21
  #torch.backends.cuda.preferred_linalg_library="cusolver"
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"
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",
33
+ #token=HF_TOKEN,
34
  transformer=None,
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)
42
 
43
  @spaces.GPU(duration=80)
44
  def generate_video(
 
96
  description="Generate video from image with LTX-Image-to-Video.",
97
  )
98
 
99
+ iface.launch(share = True)