Trkkk commited on
Commit
e9802b9
·
verified ·
1 Parent(s): 901464c

End of training

Browse files
checkpoint-1500/optimizer.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:5b7c1f678d70e2158dcff06b6c92b95b4c203d6d691d922612d4f86ff78d319f
3
  size 6876750164
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:d2a8e3896fd36c1a7a1f59dd8ac582baab9b2febb2696a65b65baa7ee3aabef8
3
  size 6876750164
checkpoint-1500/random_states_0.pkl CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:6f1efc1852541aadb00bcea7a6374a69a2f3425d524790f79f9db9867f7767a4
3
- size 14408
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:c9b6acd93bf6d61397913a26a0c0ebcc054d7388517ee87c0dd5d4db3703233b
3
+ size 14344
checkpoint-1500/unet/diffusion_pytorch_model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:2cb00417f20fef98245b3ff0f042ac2d303fe98c37fef16630240097f3037753
3
  size 3438167536
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:590bf8ee2dee6349a3f348ef389ffc278468609ee95f7af2f879833a3f23d4c9
3
  size 3438167536
checkpoint-1500/unet_ema/diffusion_pytorch_model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:00917696ec5061b4679258e35d42fb7084b73e95a762686595f6a9d949e06074
3
  size 3438167536
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:d6f0ed9d8f70206585f5c5fe4b7471e32db4cb22cb8c9cf5fe3334d291a7e786
3
  size 3438167536
logs/text2image-fine-tune/1729900372.1594048/events.out.tfevents.1729900372.IPEG-WS-008.14604.1 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:3e18a752380459adfc0ce1883d1ca5c028227c4819513f208dcb9f930d0b5233
3
+ size 2570
logs/text2image-fine-tune/1729900372.1956794/hparams.yml ADDED
@@ -0,0 +1,55 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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: null
7
+ caption_column: caption
8
+ center_crop: true
9
+ checkpointing_steps: 500
10
+ checkpoints_total_limit: null
11
+ dataloader_num_workers: 0
12
+ dataset_config_name: null
13
+ dataset_name: Trkkk/text_to_img_street_scene
14
+ dream_detail_preservation: 1.0
15
+ dream_training: false
16
+ enable_xformers_memory_efficient_attention: false
17
+ foreach_ema: false
18
+ gradient_accumulation_steps: 4
19
+ gradient_checkpointing: true
20
+ hub_model_id: Trkkk/stable_diffusion
21
+ hub_token: null
22
+ image_column: image
23
+ input_perturbation: 0
24
+ learning_rate: 1.0e-05
25
+ local_rank: -1
26
+ logging_dir: logs
27
+ lr_scheduler: constant
28
+ lr_warmup_steps: 0
29
+ max_grad_norm: 1.0
30
+ max_train_samples: null
31
+ max_train_steps: 1500
32
+ mixed_precision: fp16
33
+ noise_offset: 0
34
+ non_ema_revision: null
35
+ num_train_epochs: 24
36
+ offload_ema: false
37
+ output_dir: ./fine_tuned_model
38
+ prediction_type: null
39
+ pretrained_model_name_or_path: CompVis/stable-diffusion-v1-4
40
+ push_to_hub: true
41
+ random_flip: true
42
+ report_to: tensorboard
43
+ resolution: 256
44
+ resume_from_checkpoint: null
45
+ revision: null
46
+ scale_lr: false
47
+ seed: null
48
+ snr_gamma: 5.0
49
+ tracker_project_name: text2image-fine-tune
50
+ train_batch_size: 1
51
+ train_data_dir: null
52
+ use_8bit_adam: false
53
+ use_ema: true
54
+ validation_epochs: 5
55
+ variant: null
logs/text2image-fine-tune/events.out.tfevents.1729900372.IPEG-WS-008.14604.0 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:3b2b42adc1749e3356769d54d1f5ba8e6781ea97e80590137aa688b63e553b7a
3
+ size 2324857
unet/diffusion_pytorch_model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:00917696ec5061b4679258e35d42fb7084b73e95a762686595f6a9d949e06074
3
  size 3438167536
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:d6f0ed9d8f70206585f5c5fe4b7471e32db4cb22cb8c9cf5fe3334d291a7e786
3
  size 3438167536
val_imgs_grid.png CHANGED