Skip to content

Commit 5562c05

Browse files
committed
update RL code
1 parent 5cc5599 commit 5562c05

File tree

2 files changed

+2
-2
lines changed

2 files changed

+2
-2
lines changed

examples/OmniGen2-RL/options/omnigen2_edit_rl_4machine_editscore7b_avg4.yml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -88,7 +88,7 @@ train:
8888
num_update_steps_per_sampling: 2
8989
batch_size_per_forward: 9
9090
num_images_per_prompt: 12
91-
sigma_coef: 0.9
91+
sigma_coef: 0.7
9292
negative_prompt: ""
9393
num_inference_step: 20
9494
max_sequence_length: 1024

examples/OmniGen2-RL/options/omnigen2_edit_rl_single_machine_editscore7b.yml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -88,7 +88,7 @@ train:
8888
num_update_steps_per_sampling: 2
8989
batch_size_per_forward: 9
9090
num_images_per_prompt: 12
91-
sigma_coef: 0.9
91+
sigma_coef: 0.7
9292
negative_prompt: ""
9393
num_inference_step: 20
9494
max_sequence_length: 1024

0 commit comments

Comments
 (0)