diff --git a/BatchFuncs.py b/BatchFuncs.py index eeb93b3..a2da96f 100644 --- a/BatchFuncs.py +++ b/BatchFuncs.py @@ -173,7 +173,7 @@ def BatchPoolAnimConditioning(cur_prompt_series, nxt_prompt_series, weight_serie intermediate_cond = [] final_pooled_output = torch.cat(pooled_out, dim=0) - final_conditioning = torch.cat(cond_out, dim=1) + final_conditioning = torch.cat(cond_out, dim=0) return [[final_conditioning, {"pooled_output": final_pooled_output}]] @@ -200,7 +200,7 @@ def BatchGLIGENConditioning(cur_prompt_series, nxt_prompt_series, weight_series, cond_out.append(interpolated_cond) final_pooled_output = torch.cat(pooled_out, dim=0) - final_conditioning = torch.cat(cond_out, dim=1) + final_conditioning = torch.cat(cond_out, dim=0) return cond_out, pooled_out @@ -421,4 +421,4 @@ def BatchInterpolatePromptsSDXL(animation_promptsG, animation_promptsL, max_fram # else: # next_cond = SDXLencode(clip, width, height, crop_w, crop_h, target_width, target_height, # cur_prompt_series_G[current_frame], cur_prompt_series_L[current_frame]) -# return addWeighted(current_cond, next_cond, weight_series[current_frame]) \ No newline at end of file +# return addWeighted(current_cond, next_cond, weight_series[current_frame])