Skiittoo commited on
Commit
953280e
1 Parent(s): 2ef5356

End of training

Browse files
README.md CHANGED
@@ -14,7 +14,7 @@ inference: true
14
 
15
  # Text-to-image finetuning - Skiittoo/sd-cartoon-faces-model
16
 
17
- This pipeline was finetuned from **CompVis/stable-diffusion-v1-4** on the **Skiittoo/cartoon-faces** dataset. Below are some example images generated with the finetuned pipeline using the following prompts: ['[Cartoon face with white skin]']:
18
 
19
  ![val_imgs_grid](./val_imgs_grid.png)
20
 
@@ -28,7 +28,7 @@ from diffusers import DiffusionPipeline
28
  import torch
29
 
30
  pipeline = DiffusionPipeline.from_pretrained("Skiittoo/sd-cartoon-faces-model", torch_dtype=torch.float16)
31
- prompt = "[Cartoon face with white skin]"
32
  image = pipeline(prompt).images[0]
33
  image.save("my_image.png")
34
  ```
 
14
 
15
  # Text-to-image finetuning - Skiittoo/sd-cartoon-faces-model
16
 
17
+ This pipeline was finetuned from **CompVis/stable-diffusion-v1-4** on the **Skiittoo/cartoon-faces** dataset. Below are some example images generated with the finetuned pipeline using the following prompts: ['[Cartoon face with sunglasses]']:
18
 
19
  ![val_imgs_grid](./val_imgs_grid.png)
20
 
 
28
  import torch
29
 
30
  pipeline = DiffusionPipeline.from_pretrained("Skiittoo/sd-cartoon-faces-model", torch_dtype=torch.float16)
31
+ prompt = "[Cartoon face with sunglasses]"
32
  image = pipeline(prompt).images[0]
33
  image.save("my_image.png")
34
  ```
logs/text2image-fine-tune/1696893653.6250188/events.out.tfevents.1696893653.aa1d505ac963.147806.1 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:a2611a1fea0626d17995aa4c7d2201b71b6f6f86a6bcd75301c92150c81624ce
3
+ size 2364
logs/text2image-fine-tune/1696893653.6294377/hparams.yml ADDED
@@ -0,0 +1,50 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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: text
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: Skiittoo/cartoon-faces
14
+ enable_xformers_memory_efficient_attention: false
15
+ gradient_accumulation_steps: 4
16
+ gradient_checkpointing: true
17
+ hub_model_id: null
18
+ hub_token: null
19
+ image_column: image
20
+ input_perturbation: 0
21
+ learning_rate: 1.0e-05
22
+ local_rank: -1
23
+ logging_dir: logs
24
+ lr_scheduler: constant
25
+ lr_warmup_steps: 0
26
+ max_grad_norm: 1.0
27
+ max_train_samples: null
28
+ max_train_steps: 15000
29
+ mixed_precision: fp16
30
+ noise_offset: 0
31
+ non_ema_revision: null
32
+ num_train_epochs: 6
33
+ output_dir: sd-cartoon-faces-model
34
+ prediction_type: null
35
+ pretrained_model_name_or_path: CompVis/stable-diffusion-v1-4
36
+ push_to_hub: true
37
+ random_flip: true
38
+ report_to: tensorboard
39
+ resolution: 512
40
+ resume_from_checkpoint: /content/sd-cartoon-faces-model/checkpoint-15000
41
+ revision: null
42
+ scale_lr: false
43
+ seed: null
44
+ snr_gamma: null
45
+ tracker_project_name: text2image-fine-tune
46
+ train_batch_size: 1
47
+ train_data_dir: null
48
+ use_8bit_adam: false
49
+ use_ema: true
50
+ validation_epochs: 5
logs/text2image-fine-tune/1696894286.4379585/events.out.tfevents.1696894286.aa1d505ac963.150539.1 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:99e65bf7cba9647b4ff5a55d1cff0a4a065f5a2525778b2f50decbcc7919cfe4
3
+ size 2364
logs/text2image-fine-tune/1696894286.4397793/hparams.yml ADDED
@@ -0,0 +1,50 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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: text
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: Skiittoo/cartoon-faces
14
+ enable_xformers_memory_efficient_attention: false
15
+ gradient_accumulation_steps: 4
16
+ gradient_checkpointing: true
17
+ hub_model_id: null
18
+ hub_token: null
19
+ image_column: image
20
+ input_perturbation: 0
21
+ learning_rate: 1.0e-05
22
+ local_rank: -1
23
+ logging_dir: logs
24
+ lr_scheduler: constant
25
+ lr_warmup_steps: 0
26
+ max_grad_norm: 1.0
27
+ max_train_samples: null
28
+ max_train_steps: 15000
29
+ mixed_precision: fp16
30
+ noise_offset: 0
31
+ non_ema_revision: null
32
+ num_train_epochs: 6
33
+ output_dir: sd-cartoon-faces-model
34
+ prediction_type: null
35
+ pretrained_model_name_or_path: CompVis/stable-diffusion-v1-4
36
+ push_to_hub: true
37
+ random_flip: true
38
+ report_to: tensorboard
39
+ resolution: 512
40
+ resume_from_checkpoint: /content/sd-cartoon-faces-model/checkpoint-15000
41
+ revision: null
42
+ scale_lr: false
43
+ seed: null
44
+ snr_gamma: null
45
+ tracker_project_name: text2image-fine-tune
46
+ train_batch_size: 1
47
+ train_data_dir: null
48
+ use_8bit_adam: false
49
+ use_ema: true
50
+ validation_epochs: 5
logs/text2image-fine-tune/events.out.tfevents.1696893653.aa1d505ac963.147806.0 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:099aa88880a744c478d818c58db0a68c42b5394d8922da5fef417ee3933cc60a
3
+ size 88
logs/text2image-fine-tune/events.out.tfevents.1696894286.aa1d505ac963.150539.0 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:995276b07488f07b1c48f7a5d0c7e2ae02382d441166733a797299145e71523e
3
+ size 88
val_imgs_grid.png CHANGED