CogVideoX Disney AdamW 4000 0.0003 Constant

Requiere la biblioteca Diffusers. Carga CogVideoX-5B, incorpora los pesos LoRA y genera el vídeo a partir de una descripción textual:

import torch
from diffusers import CogVideoXPipeline
from diffusers.utils import export_to_video

pipe = CogVideoXPipeline.from_pretrained(
    "THUDM/CogVideoX-5b",
    torch_dtype=torch.bfloat16
).to("cuda")

pipe.load_lora_weights(
    "a-r-r-o-w/cogvideox-disney-adamw-4000-0.0003-constant",
    weight_name="pytorch_lora_weights.safetensors",
    adapter_name="cogvideox-lora"
)

# La ponderación refleja los valores de entrenamiento:
# lora_alpha = 32 y rank = 64.
pipe.set_adapters(["cogvideox-lora"], [32 / 64])

video = pipe(
    "BW_STYLE A black and white animated scene unfolds with an anthropomorphic goat surrounded by musical notes and symbols, suggesting a playful environment. Mickey Mouse appears, leaning forward in curiosity as the goat remains still. The goat then engages with Mickey, who bends down to converse or react. The dynamics shift as Mickey grabs the goat, potentially in surprise or playfulness, amidst a minimalistic background. The scene captures the evolving relationship between the two characters in a whimsical, animated setting, emphasizing their interactions and emotions",
    guidance_scale=6,
    use_dynamic_cfg=True
).frames[0]

export_to_video(video, "output.mp4", fps=8)