Instructions to use thu-ml/CogVideoX-RIFLEx-diffusers with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Diffusers
How to use thu-ml/CogVideoX-RIFLEx-diffusers with Diffusers:
pip install -U diffusers transformers accelerate
import torch from diffusers import DiffusionPipeline # switch to "mps" for apple devices pipe = DiffusionPipeline.from_pretrained("thu-ml/CogVideoX-RIFLEx-diffusers", dtype=torch.bfloat16, device_map="cuda") prompt = "Astronaut in a jungle, cold color palette, muted colors, detailed, 8k" image = pipe(prompt).images[0] - Notebooks
- Google Colab
- Kaggle
File size: 348 Bytes
96df135 8994f53 96df135 8994f53 | 1 2 3 4 5 6 7 8 9 10 | ---
license: apache-2.0
pipeline_tag: text-to-video
library_name: diffusers
---
This repository contains the model described in the paper [RIFLEx: A Free Lunch for Length Extrapolation in Video Diffusion Transformers](https://huggingface.co/papers/2502.15894).
Project page: https://riflex-video.github.io/.
Code: https://github.com/thu-ml/RIFLEx |