aaronhandoko01 commited on
Commit
d603c50
·
verified ·
1 Parent(s): c102dcc

End of training

Browse files
README.md ADDED
@@ -0,0 +1,43 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ base_model: stabilityai/stable-diffusion-2
3
+ library_name: diffusers
4
+ license: creativeml-openrail-m
5
+ inference: true
6
+ instance_prompt: a photo of MRI Prostate cancer Images
7
+ tags:
8
+ - text-to-image
9
+ - diffusers
10
+ - lora
11
+ - diffusers-training
12
+ - stable-diffusion
13
+ - stable-diffusion-diffusers
14
+ ---
15
+
16
+ <!-- This model card has been generated automatically according to the information the training script had access to. You
17
+ should probably proofread and complete it, then remove this comment. -->
18
+
19
+
20
+ # LoRA DreamBooth - aaronhandoko01/output_diffuex
21
+
22
+ These are LoRA adaption weights for stabilityai/stable-diffusion-2. The weights were trained on a photo of MRI Prostate cancer Images using [DreamBooth](https://dreambooth.github.io/). You can find some example images in the following.
23
+
24
+
25
+
26
+ LoRA for the text encoder was enabled: False.
27
+
28
+
29
+ ## Intended uses & limitations
30
+
31
+ #### How to use
32
+
33
+ ```python
34
+ # TODO: add an example code snippet for running this diffusion pipeline
35
+ ```
36
+
37
+ #### Limitations and bias
38
+
39
+ [TODO: provide examples of latent issues and potential remediations]
40
+
41
+ ## Training details
42
+
43
+ [TODO: describe the data used to train the model]
logs/dreambooth-lora/1742441874.8314006/events.out.tfevents.1742441874.2f064f7f2e58 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:0afcb125282749919d2397b2ac91690ecad3e60ac4aafc9e86fc0f9e44880dd8
3
+ size 2849
logs/dreambooth-lora/1742441874.8354921/hparams.yml ADDED
@@ -0,0 +1,59 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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
+ center_crop: false
7
+ checkpointing_steps: 500
8
+ checkpoints_total_limit: null
9
+ class_data_dir: null
10
+ class_labels_conditioning: null
11
+ class_prompt: null
12
+ dataloader_num_workers: 0
13
+ diffuex: true
14
+ enable_xformers_memory_efficient_attention: false
15
+ gradient_accumulation_steps: 1
16
+ gradient_checkpointing: false
17
+ hub_model_id: null
18
+ hub_token: null
19
+ image_classifier_path: /code/diffuex/diffusers/examples/dreambooth/models/epoch:76-train_loss:0.14859-val_loss:0.31609-train_acc:0.94366-val_ap:0.49077-accuracy:0.87973-F1:0.65990.pth
20
+ image_model: efficientnet-b0
21
+ instance_data_dir: /code/diffuex/dataset/diffuex_dataset/test/
22
+ instance_prompt: a photo of MRI Prostate cancer Images
23
+ learning_rate: 1.0e-06
24
+ local_rank: -1
25
+ logging_dir: logs
26
+ lr_num_cycles: 1
27
+ lr_power: 1.0
28
+ lr_scheduler: constant
29
+ lr_warmup_steps: 0
30
+ max_grad_norm: 1.0
31
+ max_train_steps: 400
32
+ mixed_precision: null
33
+ num_class_images: 100
34
+ num_train_epochs: 67
35
+ num_validation_images: 4
36
+ output_dir: /code/diffuex/diffusers/examples/dreambooth/output/output_diffuex
37
+ pre_compute_text_embeddings: false
38
+ pretrained_model_name_or_path: stabilityai/stable-diffusion-2
39
+ prior_generation_precision: null
40
+ prior_loss_weight: 1.0
41
+ push_to_hub: true
42
+ rank: 4
43
+ report_to: tensorboard
44
+ resolution: 64
45
+ resume_from_checkpoint: null
46
+ revision: null
47
+ sample_batch_size: 4
48
+ scale_lr: false
49
+ seed: null
50
+ text_encoder_use_attention_mask: false
51
+ tokenizer_max_length: null
52
+ tokenizer_name: null
53
+ train_batch_size: 1
54
+ train_text_encoder: false
55
+ use_8bit_adam: false
56
+ validation_epochs: 50
57
+ validation_prompt: null
58
+ variant: fp16
59
+ with_prior_preservation: false
logs/dreambooth-lora/events.out.tfevents.1742441874.2f064f7f2e58 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:012da3841e0f862143948c02ca53b3061081503b7ed51219361a0310d67197da
3
+ size 33302
pytorch_lora_weights.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:f3871937ceafa608d49eaf371bc1b1af3a7eaac7932fe69e6d268b4f4bb2639d
3
+ size 3357296