Hugging Face's logo

rohith2812
/
atoi-finetuned-model

Text-to-Image
Diffusers
TensorBoard
Safetensors
StableDiffusionPipeline
stable-diffusion
stable-diffusion-diffusers
diffusers-training
Inference Endpoints
Model card Files Metrics Community
rohith2812 commited on
Commit
0518627
•
1 Parent(s): 92fbd14

End of training

Browse files
Files changed (10) hide show
  1. README.md +5 -0
  2. checkpoint-100/optimizer.bin +1 -1
  3. checkpoint-100/random_states_0.pkl +1 -1
  4. checkpoint-100/unet/diffusion_pytorch_model.safetensors +1 -1
  5. checkpoint-100/unet_ema/diffusion_pytorch_model.safetensors +1 -1
  6. logs/atoi_training/text2image-fine-tune/1731962266.8380508/events.out.tfevents.1731962266.52623d98689a.8999.1 +3 -0
  7. logs/atoi_training/text2image-fine-tune/1731962266.8397677/hparams.yml +55 -0
  8. logs/atoi_training/text2image-fine-tune/events.out.tfevents.1731962266.52623d98689a.8999.0 +3 -0
  9. unet/diffusion_pytorch_model.safetensors +1 -1
  10. val_imgs_grid.png +0 -0
README.md CHANGED
@@ -9,6 +9,11 @@ tags:
9
  - text-to-image
10
  - diffusers
11
  - diffusers-training
 
 
 
 
 
12
  ---
13
 
14
  <!-- This model card has been generated automatically according to the information the training script had access to. You
 
9
  - text-to-image
10
  - diffusers
11
  - diffusers-training
12
+ - stable-diffusion
13
+ - stable-diffusion-diffusers
14
+ - text-to-image
15
+ - diffusers
16
+ - diffusers-training
17
  ---
18
 
19
  <!-- This model card has been generated automatically according to the information the training script had access to. You
checkpoint-100/optimizer.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:fcc6fe6a299602841aee53825d92be59ad0d0774a39db00f356302bc9d64d878
3
  size 6876750164
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:44f40b31b669cbf8b7406460aca90c52ec3d83b44a24cf21a9eba2b383d970cb
3
  size 6876750164
checkpoint-100/random_states_0.pkl CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:1aa49cfe94fbfc326b41524874ccef2e7e3aa407ed5b47e5978dcd1ea15b75b0
3
  size 14344
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:64dcaae95e7da968a11c1e491f8a0dfb9a038a1ab493bccc6a45d1321cde268a
3
  size 14344
checkpoint-100/unet/diffusion_pytorch_model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:ecd3303369615435dbac283d45c3857e43e333abb9d6aa28995752a13ed80826
3
  size 3438167536
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:070a271be1daf38ece04691a725aa8d57982fb41debc8c76ce81d12cc6dc278b
3
  size 3438167536
checkpoint-100/unet_ema/diffusion_pytorch_model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:680df30c2eef4b1c7575b601e5ad0f00539f4c7ddb9aeb542f1ffa3e1df2d73e
3
  size 3438167536
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:cd177ff73b2522c1b70e31070c1e34aabcfd2fa3081b32507bea57b6693630d6
3
  size 3438167536
logs/atoi_training/text2image-fine-tune/1731962266.8380508/events.out.tfevents.1731962266.52623d98689a.8999.1 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:bc4fb7f46f178e2232b2bf49a394a9072c59d25ab50110a0916dac853e9428fb
3
+ size 2506
logs/atoi_training/text2image-fine-tune/1731962266.8397677/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: combined_text
8
+ center_crop: true
9
+ checkpointing_steps: 100
10
+ checkpoints_total_limit: null
11
+ dataloader_num_workers: 0
12
+ dataset_config_name: null
13
+ dataset_name: rohith2812/atoi-finetuning-v2
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: null
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/atoi_training
27
+ lr_scheduler: constant
28
+ lr_warmup_steps: 0
29
+ max_grad_norm: 1.0
30
+ max_train_samples: null
31
+ max_train_steps: 100
32
+ mixed_precision: fp16
33
+ noise_offset: 0
34
+ non_ema_revision: null
35
+ num_train_epochs: 1
36
+ offload_ema: false
37
+ output_dir: atoi-finetuned-model
38
+ prediction_type: null
39
+ pretrained_model_name_or_path: stable-diffusion-v1-5/stable-diffusion-v1-5
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: null
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/atoi_training/text2image-fine-tune/events.out.tfevents.1731962266.52623d98689a.8999.0 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:c5ae3a0eaa41deaf99e9d80907aa41d6ea3875827898baedd6ec68a9f9e487e4
3
+ size 525589
unet/diffusion_pytorch_model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:680df30c2eef4b1c7575b601e5ad0f00539f4c7ddb9aeb542f1ffa3e1df2d73e
3
  size 3438167536
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:cd177ff73b2522c1b70e31070c1e34aabcfd2fa3081b32507bea57b6693630d6
3
  size 3438167536
val_imgs_grid.png CHANGED