import torch
from diffusers import DiffusionPipeline
from diffusers.utils import load_image, export_to_video
# switch to "mps" for apple devices
pipe = DiffusionPipeline.from_pretrained("Disty0/Index-anisora-5B_RL-diffusers", dtype=torch.bfloat16, device_map="cuda")
pipe.to("cuda")
prompt = "A man with short gray hair plays a red electric guitar."
image = load_image(
"https://huggingface.co/datasets/huggingface/documentation-images/resolve/main/diffusers/guitar-man.png"
)
output = pipe(image=image, prompt=prompt).frames[0]
export_to_video(output, "output.mp4")Diffusers version of IndexTeam/Index-anisora 5B
Source file: https://huggingface.co/IndexTeam/Index-anisora/blob/main/5B_RL/1000/mp_rank_00_model_states.pt
Soruce file sha256sum: d99db12f5942d67b711a789b47778c7db1273f109dbda18904b94628b50aaa9c
Text Encoder and VAE are from THUDM/CogVideoX-5b-I2V
- Downloads last month
- 4
Model tree for Disty0/Index-anisora-5B_RL-diffusers
Base model
IndexTeam/Index-anisora