Update BatchFuncs.py

Switched dim to 0 again to fix issues with AniDiff
This commit is contained in:
FizzleDorf
2023-10-26 13:27:10 -04:00
committed by GitHub
parent 6c62b4d38c
commit 3abf4b489c
+3 -3
View File
@@ -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])
# return addWeighted(current_cond, next_cond, weight_series[current_frame])