模型描述

这是对Mochi-1预览模型genmo/mochi-1-preview进行LoRA微调的版本。

该模型使用CogVideoX工厂进行训练——这是一个包含针对CogVideoX和Mochi系列模型的内存优化训练脚本的存储库，利用了TorchAO和DeepSpeed。这些脚本改编自CogVideoX Diffusers训练器。

下载模型

在“文件与版本”选项卡中下载LoRA。

使用方法

需要安装🧨 Diffusers库。

from diffusers import MochiPipeline
from diffusers.utils import export_to_video
import torch 

pipe = MochiPipeline.from_pretrained("genmo/mochi-1-preview")
pipe.load_lora_weights("CHANGE_ME")
pipe.enable_model_cpu_offload()

with torch.autocast("cuda", torch.bfloat16):
    video = pipe(
        prompt="CHANGE_ME",
        guidance_scale=6.0,
        num_inference_steps=64,
        height=480,
        width=848,
        max_sequence_length=256,
        output_type="np"
    ).frames[0]
export_to_video(video)

更多细节，包括权重调整、合并和融合LoRAs，请查看关于在diffusers中加载LoRAs的文档。