user029182 commited on
Commit
59b5bd8
·
verified ·
1 Parent(s): f53286d

Update video_model.py

Browse files
Files changed (1) hide show
  1. video_model.py +1 -1
video_model.py CHANGED
@@ -1,6 +1,6 @@
1
  import torch
2
 
3
- from diffusers import LTXPipeline, LTXImageToVideoPipeline, LTXVideoTransformer3DModel, GGUFQuantizationConfig
4
 
5
 
6
  device = "cuda" if torch.cuda.is_available() else "cpu"
 
1
  import torch
2
 
3
+ from diffusers import LTXPipeline, LTXImageToVideoPipeline, LTXVideoTransformer3DModel, GGUFQuantizationConfig, AutoencoderKLLTXVideo
4
 
5
 
6
  device = "cuda" if torch.cuda.is_available() else "cpu"