1inkusFace commited on
Commit
dedb2f9
·
verified ·
1 Parent(s): ada7afb

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +7 -5
app.py CHANGED
@@ -33,14 +33,16 @@ single_file_url = "https://huggingface.co/Lightricks/LTX-Video/ltx-video-2b-v0.9
33
  #vaeX = AutoencoderKLLTXVideo.from_pretrained("Lightricks/LTX-Video",subfolder='vae',token=HF_TOKEN)
34
 
35
  pipe = LTXImageToVideoPipeline.from_pretrained(
36
- "Lightricks/LTX-Video",
 
37
  token=HF_TOKEN,
38
- transformer=None,
39
  text_encoder=None,
40
  ).to(torch.device("cuda"),torch.bfloat16)
41
 
42
- text_encoder = T5EncoderModel.from_pretrained("Lightricks/LTX-Video",subfolder='text_encoder',token=True).to(torch.device("cuda"),torch.bfloat16)
43
- transformer = LTXVideoTransformer3DModel.from_single_file(single_file_url,token=HF_TOKEN).to(torch.device("cuda"),torch.bfloat16)
 
44
 
45
  @spaces.GPU(duration=80)
46
  def generate_video(
@@ -56,7 +58,7 @@ def generate_video(
56
  progress=gr.Progress(track_tqdm=True)
57
  ):
58
  pipe.text_encoder=text_encoder
59
- pipe.transformer=transformer
60
  seed=random.randint(0, MAX_SEED)
61
  generator = torch.Generator(device="cuda").manual_seed(seed)
62
  image = Image.open(image_url).convert("RGB")
 
33
  #vaeX = AutoencoderKLLTXVideo.from_pretrained("Lightricks/LTX-Video",subfolder='vae',token=HF_TOKEN)
34
 
35
  pipe = LTXImageToVideoPipeline.from_pretrained(
36
+ #"Lightricks/LTX-Video",
37
+ 'a-r-r-o-w/LTX-Video-0.9.1-diffusers',
38
  token=HF_TOKEN,
39
+ #transformer=None,
40
  text_encoder=None,
41
  ).to(torch.device("cuda"),torch.bfloat16)
42
 
43
+ #text_encoder = T5EncoderModel.from_pretrained("Lightricks/LTX-Video",subfolder='text_encoder',token=True).to(torch.device("cuda"),torch.bfloat16)
44
+ text_encoder = T5EncoderModel.from_pretrained("a-r-r-o-w/LTX-Video-0.9.1-diffusers",subfolder='text_encoder',token=True).to(torch.device("cuda"),torch.bfloat16)
45
+ #transformer = LTXVideoTransformer3DModel.from_single_file(single_file_url,token=HF_TOKEN).to(torch.device("cuda"),torch.bfloat16)
46
 
47
  @spaces.GPU(duration=80)
48
  def generate_video(
 
58
  progress=gr.Progress(track_tqdm=True)
59
  ):
60
  pipe.text_encoder=text_encoder
61
+ # pipe.transformer=transformer
62
  seed=random.randint(0, MAX_SEED)
63
  generator = torch.Generator(device="cuda").manual_seed(seed)
64
  image = Image.open(image_url).convert("RGB")