Update BatchFuncs.py
Switched dim to 0 again to fix issues with AniDiff
This commit is contained in:
+3
-3
@@ -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])
|
||||
|
||||
Reference in New Issue
Block a user