theluyuan/DiffSynth-Studio
1
0
Fork 0
mirror of https://github.com/modelscope/DiffSynth-Studio.git synced 2026-03-22 00:38:11 +00:00
Code Issues Packages Projects Releases Wiki Activity

update examples

Browse Source
This commit is contained in:
Artiprocher
2024-08-17 17:51:31 +08:00
parent 99e11112a7
commit 80aa4d8e19
5 changed files with 40 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
diffsynth/pipelines/dancer.py
View File

@@ -136,6 +136,10 @@ def lets_dance_xl(
device = "cuda",
vram_limit_level = 0,
):
# 0. Text embedding alignment (only for video processing)
if encoder_hidden_states.shape[0] != sample.shape[0]:
encoder_hidden_states = encoder_hidden_states.repeat(sample.shape[0], 1, 1, 1)
# 1. ControlNet
controlnet_insert_block_id = 22
if controlnet is not None and controlnet_frames is not None: