linluqiu commited on
Commit
ae2da3d
·
verified ·
1 Parent(s): 55f09ee

Upload folder using huggingface_hub

Browse files
Files changed (3) hide show
  1. .gitattributes +1 -0
  2. checkpoint_19017 +3 -0
  3. config.yml +50 -0
.gitattributes CHANGED
@@ -33,3 +33,4 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
33
  *.zip filter=lfs diff=lfs merge=lfs -text
34
  *.zst filter=lfs diff=lfs merge=lfs -text
35
  *tfevents* filter=lfs diff=lfs merge=lfs -text
 
 
33
  *.zip filter=lfs diff=lfs merge=lfs -text
34
  *.zst filter=lfs diff=lfs merge=lfs -text
35
  *tfevents* filter=lfs diff=lfs merge=lfs -text
36
+ checkpoint_19017 filter=lfs diff=lfs merge=lfs -text
checkpoint_19017 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:c593da8527f0a207568b7e81c7f39011f5a6f262df909db61b4d9c7055e7f8c2
3
+ size 857001698
config.yml ADDED
@@ -0,0 +1,50 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ data_path: embedded-language-flows/openwebtext-t5
2
+ max_length: 1024
3
+ encoder_model_name: t5-small
4
+ latent_mean: 0.0
5
+ latent_std: 0.2
6
+ model: ELF-B
7
+ bottleneck_dim: 128
8
+ num_time_tokens: 4
9
+ num_self_cond_cfg_tokens: 4
10
+ num_model_mode_tokens: 4
11
+ denoiser_p_mean: -1.5
12
+ denoiser_p_std: 0.8
13
+ denoiser_noise_scale: 2.0
14
+ t_eps: 0.05
15
+ time_schedule: logit_normal
16
+ decoder_prob: 0.2
17
+ decoder_noise_scale: 5.0
18
+ decoder_p_mean: 0.8
19
+ decoder_p_std: 0.8
20
+ self_cond_prob: 0.5
21
+ teacher_path: embedded-language-flows/ELF-B-owt-torch
22
+ initialize_student_from_teacher: true
23
+ teacher_use_ema: true
24
+ student_init_from_checkpoint: null
25
+ epochs: 1
26
+ global_batch_size: 512
27
+ blr: 0.001
28
+ weight_decay: 0.0
29
+ warmup_epochs: 0.1
30
+ warmup_steps: -1
31
+ optimizer: muon
32
+ ema_decay1: 0.9999
33
+ sampling_configs_path: src/configs/sampling_configs/uncond_fewstep_sampling_configs.yml
34
+ num_samples: 1000
35
+ log_freq: 100
36
+ save_freq: 1
37
+ eval_freq: 1
38
+ output_dir: outputs/elf_b_pd-owt/round5_1step
39
+ hf_repo_id: null
40
+ resume: null
41
+ use_wandb: true
42
+ wandb_project: elf
43
+ wandb_entity: null
44
+ wandb_run_name: elf_b_pd-owt_round5_1step
45
+ seed: 42
46
+ pd_step_size: 1.0
47
+ pd_student_schedule: logit_normal
48
+ pd_teacher_schedule: logit_normal
49
+ batch_size: 16
50
+ lr: 0.002