revert sdxl change due to high vram usage

2023-10-24 00:07:19 -04:00
parent 05b0aeaa05
commit 1edaaf4d7b
1 changed files with 1 additions and 1 deletions
@@ -207,7 +207,7 @@ def BatchPoolAnimConditioningSDXL(cur_prompt_series, nxt_prompt_series, weight_s
        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 [[final_conditioning, {"pooled_output": final_pooled_output}]]