imthanhlv commited on
Commit
f4d0dd9
·
verified ·
1 Parent(s): 15c34d2

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -2
app.py CHANGED
@@ -8,9 +8,8 @@ import types
8
  from typing import Optional
9
 
10
  pipe = LTXPipeline.from_pretrained("a-r-r-o-w/LTX-Video-0.9.1-diffusers", torch_dtype=torch.bfloat16)
 
11
  pipe.to("cuda")
12
- pipe.transformer = torch.compile(pipe.transformer, mode="reduce-overhead", fullgraph=True)
13
- pipe.vae = torch.compile(pipe.vae, mode="reduce-overhead", fullgraph=True)
14
 
15
 
16
  # pipe.vae.decode = vae_decode
 
8
  from typing import Optional
9
 
10
  pipe = LTXPipeline.from_pretrained("a-r-r-o-w/LTX-Video-0.9.1-diffusers", torch_dtype=torch.bfloat16)
11
+ pipe.enable_sequential_cpu_offload()
12
  pipe.to("cuda")
 
 
13
 
14
 
15
  # pipe.vae.decode = vae_decode