Update sdxl_train_controlnet.py

sdbds · Jul 8, 2024 · 1776955 · 1776955
1 parent 50022bd
commit 1776955
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/sdxl_train_controlnet.py b/sdxl_train_controlnet.py
@@ -528,7 +528,7 @@ def remove_model(old_ckpt_name):
                 #     orig_size, crop_size, target_size, accelerator.device
                 # ).to(weight_dtype)
 
-                embs = torch.cat([orig_size, crop_size, target_size], dim=-1).to(accelerator.device).to(weight_dtype)
+                embs = torch.cat([orig_size, crop_size, target_size]).to(accelerator.device).to(weight_dtype) #B,6
                 # concat embeddings
                 #vector_embedding = torch.cat([pool2, embs], dim=1).to(weight_dtype)
                 vector_embedding_dict = {