hearmeneigh commited on
Commit
67b6176
·
1 Parent(s): 8d7e6a4

End of training

Browse files
logs/text2image-fine-tune/1680550731.187756/events.out.tfevents.1680550731.07872ee260ad.19143.1 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:818c98923d9ff5b6bd2d6be55e967356f56fe8bbc213d875dd5a432a594522ac
3
+ size 2341
logs/text2image-fine-tune/1680550731.1893218/hparams.yml ADDED
@@ -0,0 +1,47 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ adam_beta1: 0.9
2
+ adam_beta2: 0.999
3
+ adam_epsilon: 1.0e-08
4
+ adam_weight_decay: 0.01
5
+ allow_tf32: false
6
+ cache_dir: /workspace/huggingface/cache
7
+ caption_column: text
8
+ center_crop: true
9
+ checkpointing_steps: 5000
10
+ checkpoints_total_limit: null
11
+ dataloader_num_workers: 0
12
+ dataset_config_name: null
13
+ dataset_name: hearmeneigh/e621-rising-v2-curated
14
+ enable_xformers_memory_efficient_attention: false
15
+ gradient_accumulation_steps: 1
16
+ gradient_checkpointing: false
17
+ hub_model_id: hearmeneigh/sd21-e621-rising-v2
18
+ hub_token: null
19
+ image_column: image
20
+ learning_rate: 4.0e-06
21
+ local_rank: 0
22
+ logging_dir: logs
23
+ lr_scheduler: constant
24
+ lr_warmup_steps: 0
25
+ maintain_aspect_ratio: true
26
+ max_grad_norm: 1.0
27
+ max_train_samples: null
28
+ max_train_steps: 1400
29
+ mixed_precision: bf16
30
+ noise_offset: 0
31
+ non_ema_revision: null
32
+ num_train_epochs: 1
33
+ output_dir: /workspace/output/sd2-hmn-v2-step-10
34
+ pretrained_model_name_or_path: /workspace/output/sd2-hmn-v2-step-9
35
+ push_to_hub: true
36
+ random_flip: true
37
+ report_to: tensorboard
38
+ resolution: 512
39
+ resume_from_checkpoint: null
40
+ revision: null
41
+ scale_lr: false
42
+ seed: null
43
+ shuffle_tags: true
44
+ train_batch_size: 34
45
+ train_data_dir: null
46
+ use_8bit_adam: false
47
+ use_ema: true
logs/text2image-fine-tune/events.out.tfevents.1680550731.07872ee260ad.19143.0 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:0ac448f6c82adc0587e20ae7f59fb3c6d7052cbb2895dbcded2c0456ea4b1c6e
3
+ size 68561
text_encoder/config.json CHANGED
@@ -1,5 +1,5 @@
1
  {
2
- "_name_or_path": "/workspace/output/sd2-hmn-v2-step-8",
3
  "architectures": [
4
  "CLIPTextModel"
5
  ],
 
1
  {
2
+ "_name_or_path": "/workspace/output/sd2-hmn-v2-step-9",
3
  "architectures": [
4
  "CLIPTextModel"
5
  ],
unet/config.json CHANGED
@@ -1,7 +1,7 @@
1
  {
2
  "_class_name": "UNet2DConditionModel",
3
  "_diffusers_version": "0.14.0",
4
- "_name_or_path": "/workspace/output/sd2-hmn-v2-step-8",
5
  "act_fn": "silu",
6
  "attention_head_dim": [
7
  5,
 
1
  {
2
  "_class_name": "UNet2DConditionModel",
3
  "_diffusers_version": "0.14.0",
4
+ "_name_or_path": "/workspace/output/sd2-hmn-v2-step-9",
5
  "act_fn": "silu",
6
  "attention_head_dim": [
7
  5,
unet/diffusion_pytorch_model.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:609f910ff977b93d2f34b05b09abdd633c41ea2d397835e598d843c247884d84
3
  size 3463934693
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:66c2ba989b9bda02da859b5d9b79e3c4be16063e718b0530f8d6faef263efb8b
3
  size 3463934693
vae/config.json CHANGED
@@ -1,7 +1,7 @@
1
  {
2
  "_class_name": "AutoencoderKL",
3
  "_diffusers_version": "0.14.0",
4
- "_name_or_path": "/workspace/output/sd2-hmn-v2-step-8",
5
  "act_fn": "silu",
6
  "block_out_channels": [
7
  128,
 
1
  {
2
  "_class_name": "AutoencoderKL",
3
  "_diffusers_version": "0.14.0",
4
+ "_name_or_path": "/workspace/output/sd2-hmn-v2-step-9",
5
  "act_fn": "silu",
6
  "block_out_channels": [
7
  128,