Hugging Face's logo

Nguyen17
/
Dev20

Text-to-Image
Diffusers
Safetensors
StableDiffusionPipeline
trl
ddpo
reinforcement-learning
stable-diffusion
Inference Endpoints
Model card Files Community
Dev20
File size: 1,514 Bytes 
575db6a
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
 
{
  "accelerator_kwargs": {},
  "allow_tf32": true,
  "async_reward_computation": false,
  "data_folder": "./inputs/selected-vincent-van-gogh",
  "exp_name": "ddpo",
  "global_step": 0,
  "high_reward": 100.0,
  "load_dataset_huggingface": false,
  "log_with": "wandb",
  "logdir": "ll",
  "low_reward": 50.0,
  "max_workers": 2,
  "mixed_precision": "fp16",
  "negative_prompts": "",
  "num_checkpoint_limit": 10,
  "num_epochs": 20,
  "offpolicy_sample_batch_size": 1,
  "per_prompt_stat_tracking": true,
  "per_prompt_stat_tracking_buffer_size": 32,
  "per_prompt_stat_tracking_min_count": 16,
  "project_kwargs": {
    "automatic_checkpoint_naming": true,
    "logging_dir": "./outputs/logs",
    "project_dir": "./outputs/",
    "total_limit": 5
  },
  "resolution": 768,
  "resume_from": "",
  "reward_function_usage": false,
  "run_name": "",
  "sample_batch_size": 2,
  "sample_eta": 1.0,
  "sample_guidance_scale": 5.0,
  "sample_num_batches_per_epoch": 1,
  "sample_num_steps": 50,
  "save_freq": 200,
  "seed": 6789,
  "tracker_kwargs": {},
  "tracker_project_name": "Dev",
  "train_adam_beta1": 0.9,
  "train_adam_beta2": 0.999,
  "train_adam_epsilon": 1e-08,
  "train_adam_weight_decay": 0.0001,
  "train_adv_clip_max": 5,
  "train_batch_size": 2,
  "train_cfg": true,
  "train_clip_range": 0.0001,
  "train_gradient_accumulation_steps": 1,
  "train_learning_rate": 0.0002,
  "train_max_grad_norm": 1.0,
  "train_num_inner_epochs": 1,
  "train_timestep_fraction": 1.0,
  "train_use_8bit_adam": false
}