2

zRzRzRzRzRzRzR · zRzRzRzRzRzRzR · commit b007be09e46f · 2025-02-20T01:10:30.000+08:00
diff --git a/examples/cogview4-control/train_control_cogview4.py b/examples/cogview4-control/train_control_cogview4.py
@@ -1066,7 +1066,7 @@ def load_model_hook(models, input_dir):
 
                 with torch.no_grad():
                     prompt_embeds, pooled_prompt_embeds, = text_encoding_pipeline.encode_prompt(
-                        captions, None
+                        captions, ""
                     )
                 original_size = (args.resolution, args.resolution)
                 original_size = torch.tensor([original_size], dtype=prompt_embeds.dtype, device=prompt_embeds.device)

Original file line number	Diff line number	Diff line change
`@@ -1066,7 +1066,7 @@ def load_model_hook(models, input_dir):`
`1066`	`1066`
`1067`	`1067`	`with torch.no_grad():`
`1068`	`1068`	`prompt_embeds, pooled_prompt_embeds, = text_encoding_pipeline.encode_prompt(`
`1069`		`- captions, None`
	`1069`	`+ captions, ""`
`1070`	`1070`	`)`
`1071`	`1071`	`original_size = (args.resolution, args.resolution)`
`1072`	`1072`	`original_size = torch.tensor([original_size], dtype=prompt_embeds.dtype, device=prompt_embeds.device)`