Skip to content

Commit 98a2417

Browse files
Update train_control_cogview4.py
1 parent fc3830c commit 98a2417

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

examples/cogview4-control/train_control_cogview4.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -132,6 +132,7 @@ def log_validation(cogview4_transformer, args, accelerator, weight_dtype, step,
132132
control_image=validation_image,
133133
num_inference_steps=50,
134134
guidance_scale=args.guidance_scale,
135+
max_sequence_length=args.max_sequence_length,
135136
generator=generator,
136137
height=args.resolution,
137138
width=args.resolution,

0 commit comments

Comments
 (0)