Skip to content

Commit 2f74c4e

Browse files
fix
1 parent fe6287a commit 2f74c4e

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

examples/cogview4-control/train_control_cogview4.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -132,7 +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=max_sequence_length, # For downstream task training usage, training can be performed on a batch basis.
135+
max_sequence_length=args.max_sequence_length, # For downstream task training usage, training can be performed on a batch basis.
136136
padding_type="max_length",
137137
generator=generator,
138138
height=args.resolution,

0 commit comments

Comments
 (0)