Update dpo_iter_parm_gudie/config.yaml
Browse files
dpo_iter_parm_gudie/config.yaml
CHANGED
@@ -5,21 +5,21 @@ wandb:
|
|
5 |
experiment:
|
6 |
project: training
|
7 |
name: show-o-dpo
|
8 |
-
output_dir:
|
9 |
save_every: 10000
|
10 |
eval_every: 2500
|
11 |
generate_every: 1000
|
12 |
log_every: 1
|
13 |
log_grad_norm_every: 20
|
14 |
-
logging_dir:
|
15 |
model:
|
16 |
-
rw_model:
|
17 |
vq_model:
|
18 |
type: magvitv2
|
19 |
vq_model_name: showlab/magvitv2
|
20 |
showo:
|
21 |
load_from_showo: false
|
22 |
-
pretrained_model_path:
|
23 |
w_clip_vit: false
|
24 |
vocab_size: 58498
|
25 |
llm_vocab_size: 50295
|
@@ -31,7 +31,7 @@ model:
|
|
31 |
dataset:
|
32 |
gen_type: t2i
|
33 |
params:
|
34 |
-
data_path:
|
35 |
validation_prompts_file: validation_prompts/showoprompts.txt
|
36 |
shuffle_buffer_size: 1000
|
37 |
num_workers: 32
|
|
|
5 |
experiment:
|
6 |
project: training
|
7 |
name: show-o-dpo
|
8 |
+
output_dir: train_set_hq_prune_full_t2i_dpo_10k_1e-5_v2_ad_orm_RM
|
9 |
save_every: 10000
|
10 |
eval_every: 2500
|
11 |
generate_every: 1000
|
12 |
log_every: 1
|
13 |
log_grad_norm_every: 20
|
14 |
+
logging_dir: train_set_hq_prune_full_t2i_dpo_10k_1e-5_v2_ad_orm_RM/logs
|
15 |
model:
|
16 |
+
rw_model: llava-onevision-google_siglip-so400m-patch14-384-Qwen_Qwen2-7B-Instruct-ov_stage_am9--train_ad_orm
|
17 |
vq_model:
|
18 |
type: magvitv2
|
19 |
vq_model_name: showlab/magvitv2
|
20 |
showo:
|
21 |
load_from_showo: false
|
22 |
+
pretrained_model_path: train_set_hq_prune_full_t2i_dpo_10k_1e-5
|
23 |
w_clip_vit: false
|
24 |
vocab_size: 58498
|
25 |
llm_vocab_size: 50295
|
|
|
31 |
dataset:
|
32 |
gen_type: t2i
|
33 |
params:
|
34 |
+
data_path: t2i_dpo_draft.yaml
|
35 |
validation_prompts_file: validation_prompts/showoprompts.txt
|
36 |
shuffle_buffer_size: 1000
|
37 |
num_workers: 32
|