Nguyen17 commited on
Commit
f7db739
·
verified ·
1 Parent(s): 38c60de

Push model using huggingface_hub.

Browse files
config.json CHANGED
@@ -9,7 +9,7 @@
9
  "mixed_precision": "fp16",
10
  "negative_prompts": "",
11
  "num_checkpoint_limit": 5,
12
- "num_epochs": 2,
13
  "offline_data_folder": "../data_offline/",
14
  "per_prompt_stat_tracking": true,
15
  "per_prompt_stat_tracking_buffer_size": 32,
@@ -36,13 +36,13 @@
36
  "train_adam_epsilon": 1e-08,
37
  "train_adam_weight_decay": 0.0001,
38
  "train_adv_clip_max": 5,
39
- "train_batch_size": 2,
40
  "train_cfg": true,
41
  "train_clip_range": 0.0001,
42
  "train_gradient_accumulation_steps": 1,
43
  "train_learning_rate": 0.0003,
44
  "train_max_grad_norm": 1.0,
45
- "train_num_inner_epochs": 1,
46
  "train_timestep_fraction": 1.0,
47
  "train_use_8bit_adam": false
48
  }
 
9
  "mixed_precision": "fp16",
10
  "negative_prompts": "",
11
  "num_checkpoint_limit": 5,
12
+ "num_epochs": 3,
13
  "offline_data_folder": "../data_offline/",
14
  "per_prompt_stat_tracking": true,
15
  "per_prompt_stat_tracking_buffer_size": 32,
 
36
  "train_adam_epsilon": 1e-08,
37
  "train_adam_weight_decay": 0.0001,
38
  "train_adv_clip_max": 5,
39
+ "train_batch_size": 4,
40
  "train_cfg": true,
41
  "train_clip_range": 0.0001,
42
  "train_gradient_accumulation_steps": 1,
43
  "train_learning_rate": 0.0003,
44
  "train_max_grad_norm": 1.0,
45
+ "train_num_inner_epochs": 2,
46
  "train_timestep_fraction": 1.0,
47
  "train_use_8bit_adam": false
48
  }
pytorch_lora_weights.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:bacd4d26f5f05c191db31369d606951d1e8f9b0bbfeb036ae3548e210d1ab54c
3
- size 3357296
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:64c7034ff2f797ab64e575bb363a44af7c8cba22c73528ecfe459602cb0c8a6d
3
+ size 5055360
unet/diffusion_pytorch_model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:61bcfee511c24736e541f32e74fd1f32af393f6cc5a910f0ce5e43d99671dbaa
3
- size 1735264088
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:58809bae310f5cd7126742f2d38ebb410b5e590ecca5f12e1eaf9eb317c80615
3
+ size 1736963552