基于EasyAnimate模型的视频生成最佳实践

  • 阿里云大数据AI技术

    2024-06-19 10:52:07
  • 人工智能
  • 原创

EasyAnimate是阿里云PAI平台自主研发的DiT的视频生成框架，它提供了完整的高清长视频生成解决方案，包括视频数据预处理、VAE训练、DiT训练、模型推理和模型评测等。本文为您介绍如何在PAI平台集成EasyAnimate并一键完成模型推理、微调及部署的实践流程。

背景信息

本文为您介绍以下两种视频生成的方式：

DSW是为算法开发者量身打造的一站式AI开发平台，集成了JupyterLab、WebIDE、Terminal多种云端开发环境，其中，Gallery提供了丰富的案例和解决方案，帮助您快速熟悉研发流程。您可以打开DSW Gallery中的案例教程，实现一键式运行Notebook，完成基于EasyAnimate的视频生成模型的推理和训练任务，也可以进行模型推理和微调等二次开发操作。

快速开始集成了众多AI开源社区中优质的预训练模型，并且基于开源模型支持零代码实现从训练到部署再到推理的全部过程，您可以通过快速开始一键部署EasyAnimate模型并生成视频，享受更快、更高效、更便捷的AI应用体验。

费用说明

  • 如果您的账号为DSW或EAS的新用户，可以免费试用相应产品。关于免费试用的额度、领取方式及注意事项等详细信息，请参见 新用户免费试用。

说明：在本教程中，仅支持使用DSW免费资源类型ecs.gn7i-c8g1.2xlarge。

前提条件

方式一：使用DSW

步骤一：创建DSW实例

  1. 进入DSW页面。

  2. 登录 PAI控制台。

  3. 在 概览页面选择目标地域。

  4. 在左侧导航栏单击 工作空间列表，在工作空间列表页面中单击目标工作空间名称，进入对应工作空间内。

  5. 在工作空间页面的左侧导航栏选择 模型开发与训练>交互式建模（DSW），进入DSW页面。

  6. 单击 创建实例。

  7. 在 配置实例向导页面，配置以下关键参数，其他参数保持默认即可。

  1. 单击 确定。

步骤二：安装EasyAnimate模型

  1. 打开DSW实例。

  2. 登录 PAI控制台。

  3. 在左侧导航栏单击 工作空间列表，在工作空间列表页面中单击目标工作空间，进入对应工作空间。

  4. 在页面左上方，选择使用服务的地域。

  5. 在左侧导航栏，选择 模型开发与训练> 交互式建模（DSW）。

  6. 单击目标DSW实例 操作列下的 打开，进入DSW实例的开发环境。

  7. 在 Notebook页签的 Launcher页面，单击 快速开始区域 Tool下的 DSW Gallery，打开DSW Gallery页面。

  8. 在DSW Gallery页面中，搜索 EasyPhoto: 基于EasyAnimate的AI视频生成示例，单击 在DSW中打开，即可自动将本教程所需的资源和教程文件下载至DSW实例中，并在下载完成后自动打开教程文件。

  1. 下载EasyAnimate相关代码和模型并进行安装。

在教程文件 easyanimate.ipynb中，单击开始按钮运行 环境安装节点命令，包括 定义函数、 下载代码和 下载模型。当成功运行一个步骤命令后，再顺次运行下个步骤的命令。

步骤三：推理模型

方式一：使用代码进行推理

单击开始按钮运行 模型推理> 使用代码进行推理节点的命令进行模型推理。您可在**/mnt/workspace/demos/easyanimate/EasyAnimate/samples/easyanimate-videos**目录中查看生成结果。您可以修改不同的参数来达到不同的效果，参数说明：

方式二：使用WebUI进行推理

  1. 单击开始按钮运行 模型推理> UI启动节点的命令，进行模型推理。

  2. 单击生成的链接，进入WebUI界面。

  1. 在WebUI界面选择预训练的模型路径、微调后的基模型和LoRA模型，设置LoRA权重，其它参数按需配置即可。

  1. 单击 Generate（生成），等待一段时间后，即可在右侧查看或下载生成的视频。

步骤四：微调LoRA

EasyAnimate提供了丰富的模型训练方式，包括DiT模型的训练（LoRA微调和基模型的全量微调）和VAE的训练。关于Gallery中内置的LoRA微调部分，更多信息，请参见 EasyAnimate。

准备数据

单击开始按钮执行 模型训练> 数据准备节点的命令，即可下载示例数据，用于模型训练。您也可以按照如下格式要求自行准备数据文件。文件数据格式如下。

project/
├── datasets/
│   ├── internal_datasets/
│       ├── videos/
│       │   ├── 00000001.mp4
│       │   ├── 00000002.mp4
│       │   └── .....
│       └── json_of_internal_datasets.json

其中，JSON文件数据格式和参数说明如下。

[
    {
      "file_path": "videos/00000001.mp4",
      "text": "A group of young men in suits and sunglasses are walking down a city street.",
      "type": "video"
    },
    {
      "file_path": "videos/00000002.mp4",
      "text": "A notepad with a drawing of a woman on it.",
      "type": "video"
    }
    .....
]

训练模型

  1. 将对应的训练脚本中的DATASET_NAME及DATASET_META_NAME设置为训练数据所在目录及训练文件地址。

export DATASET_NAME=“” # 训练数据所在目录
export DATASET_META_NAME=“datasets/Minimalism/metadata_add_width_height.json” # 训练文件地址

  1. 单击开始按钮执行 启动训练> LoRA训练节点的命令。

  2. 训练完成后，将生成的模型移动至 models/Personalized_model文件夹，即可在UI界面中选择，或单击开始按钮执行 LoRA模型推理节点的命令，指定lora_path进行视频生成。

方式二：使用快速开始

步骤一：部署模型

  1. 进入快速开始页面。

  2. 登录 PAI控制台。

  3. 在左侧导航栏单击 工作空间列表，在工作空间列表页面单击目标工作空间名称，进入对应工作空间内。

  4. 在左侧导航栏单击 快速开始，进入快速开始页面。

  5. 在快速开始页面，搜索 EasyAnimate 高清长视频生成，然后单击 部署，配置相关参数。

EasyAnimate目前仅支持使用bf16进行推理，请选择A10及其以上的显卡。

  1. 单击 部署，在弹出的 计费提醒对话框中，单击 确定，页面将自动跳转到 服务详情页面。

当 状态变为 运行中时，即表示模型部署成功。

步骤二：使用模型

模型部署完成后，您可以使用WebUI及API两种方式调用服务来生成视频。

WebUI方式

  1. 在 服务详情页面，单击 查看WEB应用。

  1. 在WebUI界面选择预训练的模型路径，其它参数按需配置即可。

  1. 单击 Generate（生成），等待一段时间后，即可在右侧查看或下载生成的视频。

API方式

  1. 在 服务详情页面的 资源详情区域，单击 查看调用信息，获取调用服务所需的信息。

  1. 通过接口更新Transformer模型，可在DSW实例或本地Python环境中执行。

如果已经在WebUI中选择模型，则无需发送请求重复调用。如遇请求超时，请在EAS日志中确认模型已加载完毕。加载完成，日志中将提示Update diffusion transformer done。Python请求示例如下。

import json
import requests
def post_diffusion_transformer(diffusion_transformer_path, url='http://127.0.0.1:7860', token=None):
    datas = json.dumps({
        "diffusion_transformer_path": diffusion_transformer_path
    })
    head = {
        'Authorization': token
    }
    r = requests.post(f'{url}/easyanimate/update_diffusion_transformer', data=datas, headers=head, timeout=15000)
    data = r.content.decode('utf-8')
    return data
def post_update_edition(edition, url='http://0.0.0.0:7860',token=None):
    head = {
        'Authorization': token
    }
    datas = json.dumps({
        "edition": edition
    })
    r = requests.post(f'{url}/easyanimate/update_edition', data=datas, headers=head)
    data = r.content.decode('utf-8')
    return data
  
if __name__ == '__main__':
    url = ''
    token = ''
    # -------------------------- #
    #  Step 1: update edition
    # -------------------------- #
    edition = "v2"
    outputs = post_update_edition(edition,url = url,token=token)
    print('Output update edition: ', outputs)
    # -------------------------- #
    #  Step 2: update edition
    # -------------------------- #
    # 默认路径不可修改
    diffusion_transformer_path = "/mnt/models/Diffusion_Transformer/EasyAnimateV2-XL-2-512x512"
    outputs = post_diffusion_transformer(diffusion_transformer_path, url = url, token=token)
    print('Output update edition: ', outputs)

其中：

  • ：替换为 步骤1中查询到的服务访问地址。

  • ：替换为 步骤1中查询到的服务Token。

  1. 调用服务，生成视频或图片。

  • 服务输入参数说明

图片

  • Python请求示例

服务返回base64_encoding，为base64结果。您可以在**/mnt/workspace/demos/easyanimate/**目录中查看生成结果。

import base64
import json
import sys
import time
from datetime import datetime
from io import BytesIO
import cv2
import requests
import base64
def post_infer(is_image, length_slider, url='http://127.0.0.1:7860',token=None):
    head = {
        'Authorization': token
    }
    datas = json.dumps({
        "base_model_path": "none",
        "motion_module_path": "none",
        "lora_model_path": "none", 
        "lora_alpha_slider": 0.55, 
        "prompt_textbox": "This video shows Mount saint helens, washington - the stunning scenery of a rocky mountains during golden hours - wide shot. A soaring drone footage captures the majestic beauty of a coastal cliff, its red and yellow stratified rock faces rich in color and against the vibrant turquoise of the sea.", 
        "negative_prompt_textbox": "Strange motion trajectory, a poor composition and deformed video, worst quality, normal quality, low quality, low resolution, duplicate and ugly, strange body structure, long and strange neck, bad teeth, bad eyes, bad limbs, bad hands, rotating camera, blurry camera, shaking camera", 
        "sampler_dropdown": "Euler", 
        "sample_step_slider": 30, 
        "width_slider": 672, 
        "height_slider": 384, 
        "is_image": is_image,
        "length_slider": length_slider,
        "cfg_scale_slider": 6,
        "seed_textbox": 43,
    })
    r = requests.post(f'{url}/easyanimate/infer_forward', data=datas, headers=head,timeout=1500)
    data = r.content.decode('utf-8')
    return data
if __name__ == '__main__':
    # initiate time
    now_date    = datetime.now()
    time_start  = time.time()  
    
    url = ''
    token = ''
    # -------------------------- #
    #  Step 3: infer
    # -------------------------- #
    is_image = False
    length_slider = 27
    outputs = post_infer(is_image, length_slider, url = url, token=token)
    
    # Get decoded data
    outputs = json.loads(outputs)
    base64_encoding = outputs["base64_encoding"]
    decoded_data = base64.b64decode(base64_encoding)
    if is_image or length_slider == 1:
        file_path = "1.png"
    else:
        file_path = "1.mp4"
    with open(file_path, "wb") as file:
        file.write(decoded_data)
        
    # End of record time
    time_end = time.time()  
    time_sum = (time_end - time_start) % 60 
    print('# --------------------------------------------------------- #')
    print(f'#   Total expenditure: {time_sum}s')
    print('# --------------------------------------------------------- #')

其中：

  • < eas-service-url>：替换为 步骤1中查询到的服务访问地址。

  • < eas-service-token>：替换为 步骤1中查询到的服务Token。


请使用浏览器的分享功能分享到微信等