15 KiB

Raw Blame History

AnimateDiff for Stable Diffusion WebUI

This extension aim for integrating AnimateDiff into AUTOMATIC1111 Stable Diffusion WebUI. You can generate GIFs in exactly the same way as generating images after enabling this extension.

This extension implements AnimateDiff in a different way. It does not require you to clone the whole SD1.5 repository. It also applied (probably) the least modification to ldm, so that you do not need to reload your model weights if you don't want to.

Batch size on WebUI will be replaced by GIF frame number internally: 1 full GIF generated in 1 batch. If you want to generate multiple GIF at once, please change batch number.

Batch number is NOT the same as batch size. In A1111 WebUI, batch number is above batch size. Batch number means the number of sequential steps, but batch size means the number of parallel steps. You do not have to worry too much when you increase batch number, but you do need to worry about your VRAM when you increase your batch size (where in this extension, video frame number). You do not need to change batch size at all when you are using this extension.

You might also be interested in another extension I created: Segment Anything for Stable Diffusion WebUI.

How to Use

Install this extension via link.
Download motion modules and put the model weights under stable-diffusion-webui/extensions/sd-webui-animatediff/model/. If you want to use another directory to save the model weights, please go to Settings/AnimateDiff. See model zoo for a list of available motion modules.
Enable Pad prompt/negative prompt to be same length and Batch cond/uncond and click Apply settings in Settings. You must do this to prevent generating two separate unrelated GIFs.

WebUI

Go to txt2img if you want to try txt2gif and img2img if you want to try img2gif.
Choose an SD1.5 checkpoint, write prompts, set configurations such as image width/height. If you want to generate multiple GIFs at once, please change batch number, instead of batch size.
Enable AnimateDiff extension, and set up each parameter, and click Generate.
1. Number of frames — Choose whatever number you like.
  
  If you enter 0 (default):
  - If you submit a video via Video source / enter a video path via Video path / enable ANY batch ControlNet, the number of frames will be the number of frames in the video (use shortest if more than one videos are submitted).
  - Otherwise, the number of frames will be your Batch size described below.
  If you enter something smaller than your Batch size other than 0: you will get the first Number of frames frames as your output GIF from your whole generation. All following frames will not appear in your generated GIF, but will be saved as PNGs as usual.
2. FPS — Frames per second, which is how many frames (images) are shown every second. If 16 frames are generated at 8 frames per second, your GIF’s duration is 2 seconds. If you submit a source video, your FPS will be the same as the source video.
3. Display loop number — How many times the GIF is played. A value of 0 means the GIF never stops playing.
4. Batch size — How many frames will be passed into the motion module at once. The model is trained with 16 frames, so it’ll give the best results when the number of frames is set to 16. Choose [1, 24] for V1 motion modules and [1, 32] for V2 motion modules.
5. Closed loop — If you enable this option and your number of frames is greater than your batch size, this extension will try to make the last frame the same as the first frame.
6. Stride — Max motion stride as a power of 2 (default: 1).
7. Overlap — Number of frames to overlap in context. If overlap is -1 (default): your overlap will be Batch size // 4.
8. Save — Format of the output. Choose at least one of "GIF"|"MP4"|"PNG". Check "TXT" if you want infotext, which will live in the same directory as the output GIF.
  1. You can optimize GIF with gifsicle (apt install gifsicle required, read #91 for more information) and/or palette (read #104 for more information). Go to Settings/AnimateDiff to enable them.
9. Reverse — Append reversed frames to your output. See #112 for instruction.
10. Frame Interpolation Interpolate between frames with Deforum's FILM implementation. Requires Deforum extension. #128
11. Interp X Replace each input frame with X interpolated output frames. #128.
12. Video source — [Optional] Video source file for video to video generation. You MUST enable ControlNet. It will be the source control for ALL ControlNet units that you enable without submitting a control image or a path to ControlNet panel. You can of course submit a control image via Single Image tab or an input directory via Batch tab, they will override this video source input and work as usual.
13. Video path — [Optional] Folder for source frames for video to video generation, but lower priority than Video source. You MUST enable ControlNet. It will be the source control for ALL ControlNet units that you enable without submitting a control image or a path to ControlNet. You can of course submit a control image via Single Image tab or an input directory via Batch tab, they will override this video path input and work as usual.
  - For people who want to inpaint videos: enter a folder which contains two sub-folders image and mask. They should contain the same number of images. This extension will match them according to the same sequence. You should not upload a video directly to Video source mentioned above. Using my Segment Anything extension can make your life much easier.
You should see the output GIF on the output gallery. You can access GIF output at stable-diffusion-webui/outputs/{txt2img or img2img}-images/AnimateDiff. You can also access image frames at stable-diffusion-webui/outputs/{txt2img or img2img}-images/{date}.

img2GIF

You need to go to img2img and submit an init frame via A1111 panel. You can optionally submit a last frame via extension panel (experiment feature, not tested, not sure if it will work).

By default: your init_latent will be changed to

init_alpha = (1 - frame_number ^ latent_power / latent_scale)
init_latent = init_latent * init_alpha + random_tensor * (1 - init_alpha)

If you upload a last frame: your init_latent will be changed in a similar way. Read this code to understand how it works.

API

Just like how you use ControlNet. Here is a sample. You will get a list of generated frames. You will have to view GIF in your file system, as mentioned at #WebUI item 4.

'alwayson_scripts': {
  'AnimateDiff': {
  	'args': [{
  		  'enable': True,         # enable AnimateDiff
  		  'video_length': 16,     # video frame number, 0-24 for v1 and 0-32 for v2
  		  'format': 'MP4',        # 'GIF' | 'MP4' | 'PNG' | 'TXT'
  		  'loop_number': 0,       # 0 = infinite loop
  		  'fps': 8,               # frames per second
  		  'model': 'mm_sd_v15_v2.ckpt',   # motion module name
  		  'reverse': [],          # 0 | 1 | 2 - 0: Add Reverse Frame, 1: Remove head, 2: Remove tail
        # parameters below are for img2gif only.
  		  'latent_power': 1,
  		  'latent_scale': 32,
  		  'last_frame': None,
  		  'latent_power_last': 1,
  		  'latent_scale_last': 32
  	  }
  	]
  }
},

Motion Module Model Zoo

mm_sd_v14.ckpt & mm_sd_v15.ckpt & mm_sd_v15_v2.ckpt by @guoyww: Google Drive | HuggingFace | CivitAI | Baidu NetDisk
mm_sd_v14.safetensors & mm_sd_v15.safetensors & mm_sd_v15_v2.safetensors by @neph1: HuggingFace
mm-Stabilized_high.pth & mm-Stabbilized_mid.pth by @manshoety: HuggingFace
temporaldiff-v1-animatediff.ckpt by @CiaraRowles: HuggingFace

Update

2023/07/20 v1.1.0: fix gif duration, add loop number, remove auto-download, remove xformers, remove instructions on gradio UI, refactor README, add sponsor QR code.
2023/07/24 v1.2.0: fix incorrect insertion of motion modules, add option to change path to save motion modules in Settings/AnimateDiff, fix loading different motion modules.
2023/09/04 v1.3.0: support any community models with the same architecture; fix grey problem via #63 (credit to @TDS4874 and @opparco)
2023/09/11 v1.4.0: support official v2 motion module (different architecture: GroupNorm not hacked, UNet middle layer has motion module).
2023/09/14: v1.4.1: always change beta, alpha_comprod and alpha_comprod_prev to resolve grey problem in other samplers.
2023/09/16: v1.5.0: randomize init latent to support better img2gif, credit to this forked repo; add other output formats and infotext output, credit to @zappityzap; add appending reversed frames; refactor code to ease maintaining.
2023/09/19: v1.5.1: support xformers, sdp, sub-quadratic attention optimization - VRAM usage decrease to 5.60GB with default setting. See FAQ 1st item for more information.
2023/09/22: v1.5.2: option to disable xformers at Settings/AnimateDiff due to a bug in xformers, API support, option to enable GIF paletter optimization at Settings/AnimateDiff (credit to @rkfg), gifsicle optimization move to Settings/AnimateDiff.
2023/09/25: v1.6.0: motion LoRA supported. Download and use them like any other LoRA you use (example: download motion lora to stable-diffusion-webui/models/Lora and add <lora:v2_lora_PanDown:0.8> to your positive prompt). Motion LoRA only supports V2 motion modules.
2023/09/27: v1.7.0: ControlNet supported. Please closely follow the instructions in How to Use, especially the explanation of Video source and Video path attributes. ControlNet is way more complex than what I can test and I ask you to test for me. Please submit an issue whenever you find a bug. You may want to check Do not append detectmap to output in Settings/ControlNet to avoid having a series of control images in your output gallery. Safetensors for some motion modules are also available now.
2023/09/29: v1.8.0: Infinite generation (with/without ControlNet) supported. Demo and video instructions are coming soon.

Prompt Travel and other CLI features are currently work in progress inside #71. Stay tuned and they should be released soon.

FAQ

Q: How much VRAM do I need?

A: Actual VRAM usage depends on your image size and video frame number. You can try to reduce image size or video frame number to reduce VRAM usage. I list some data tested on Ubuntu 22.04, NVIDIA 4090, torch 2.0.1+cu117, H=W=512, frame=16 (default setting):

Optimization VRAM usage

No optimization 12.13GB

xformers/sdp 5.60GB

sub-quadratic 10.39GB
Q: Can I use SDXL to generate GIFs?

A: You will have to wait for someone to train SDXL-specific motion modules which will have a different model architecture. This extension essentially inject multiple motion modules into SD1.5 UNet. It does not work for other variations of SD, such as SD2.1 and SDXL.