wannyansports commited on
Commit
f69b714
1 Parent(s): 88fb849

End of training

Browse files
README.md ADDED
@@ -0,0 +1,95 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ tags:
3
+ - stable-diffusion
4
+ - stable-diffusion-diffusers
5
+ - diffusers-training
6
+ - text-to-image
7
+ - diffusers
8
+ - lora
9
+ - template:sd-lora
10
+ widget:
11
+
12
+ - text: '<pokemon> with white background'
13
+ output:
14
+ url:
15
+ "image_0.png"
16
+
17
+ - text: '<pokemon> with white background'
18
+ output:
19
+ url:
20
+ "image_1.png"
21
+
22
+ - text: '<pokemon> with white background'
23
+ output:
24
+ url:
25
+ "image_2.png"
26
+
27
+ - text: '<pokemon> with white background'
28
+ output:
29
+ url:
30
+ "image_3.png"
31
+
32
+ base_model: emilianJR/epiCRealism
33
+ instance_prompt: <pokemon>
34
+ license: openrail++
35
+ ---
36
+
37
+ # SD1.5 LoRA DreamBooth - wannyansports/sdpokemon
38
+
39
+ <Gallery />
40
+
41
+ ## Model description
42
+
43
+ ### These are wannyansports/sdpokemon LoRA adaption weights for emilianJR/epiCRealism.
44
+
45
+ ## Download model
46
+
47
+ ### Use it with UIs such as AUTOMATIC1111, Comfy UI, SD.Next, Invoke
48
+
49
+ - **LoRA**: download **[`sdpokemon.safetensors` here 💾](/wannyansports/sdpokemon/blob/main/sdpokemon.safetensors)**.
50
+ - Place it on your `models/Lora` folder.
51
+ - On AUTOMATIC1111, load the LoRA by adding `<lora:sdpokemon:1>` to your prompt. On ComfyUI just [load it as a regular LoRA](https://comfyanonymous.github.io/ComfyUI_examples/lora/).
52
+ - *Embeddings*: download **[`sdpokemon_emb.safetensors` here 💾](/wannyansports/sdpokemon/blob/main/sdpokemon_emb.safetensors)**.
53
+ - Place it on it on your `embeddings` folder
54
+ - Use it by adding `sdpokemon_emb` to your prompt. For example, `<pokemon>`
55
+ (you need both the LoRA and the embeddings as they were trained together for this LoRA)
56
+
57
+
58
+ ## Use it with the [🧨 diffusers library](https://github.com/huggingface/diffusers)
59
+
60
+ ```py
61
+ from diffusers import AutoPipelineForText2Image
62
+ import torch
63
+ from huggingface_hub import hf_hub_download
64
+ from safetensors.torch import load_file
65
+
66
+ pipeline = AutoPipelineForText2Image.from_pretrained('runwayml/stable-diffusion-v1-5', torch_dtype=torch.float16).to('cuda')
67
+ pipeline.load_lora_weights('wannyansports/sdpokemon', weight_name='pytorch_lora_weights.safetensors')
68
+ embedding_path = hf_hub_download(repo_id='wannyansports/sdpokemon', filename='sdpokemon_emb.safetensors', repo_type="model")
69
+ state_dict = load_file(embedding_path)
70
+ pipeline.load_textual_inversion(state_dict["clip_l"], token=[], text_encoder=pipeline.text_encoder, tokenizer=pipeline.tokenizer)
71
+
72
+ image = pipeline('<pokemon> with white background').images[0]
73
+ ```
74
+
75
+ For more details, including weighting, merging and fusing LoRAs, check the [documentation on loading LoRAs in diffusers](https://huggingface.co/docs/diffusers/main/en/using-diffusers/loading_adapters)
76
+
77
+ ## Trigger words
78
+
79
+ To trigger image generation of trained concept(or concepts) replace each concept identifier in you prompt with the new inserted tokens:
80
+
81
+ to trigger concept `TOK` → use `<s0><s1>` in your prompt
82
+
83
+
84
+
85
+ ## Details
86
+ All [Files & versions](/wannyansports/sdpokemon/tree/main).
87
+
88
+ The weights were trained using [🧨 diffusers Advanced Dreambooth Training Script](https://github.com/huggingface/diffusers/blob/main/examples/advanced_diffusion_training/train_dreambooth_lora_sd15_advanced.py).
89
+
90
+ LoRA for the text encoder was enabled. False.
91
+
92
+ Pivotal tuning was enabled: True.
93
+
94
+ Special VAE used for training: None.
95
+
image_0.png ADDED
image_1.png ADDED
image_2.png ADDED
image_3.png ADDED
logs/dreambooth-lora-sd-15/1726467319.0785189/events.out.tfevents.1726467319.dc44792b2f5e.1319.1 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:80dc627472d8e92bc186548a8c98f6bc3dc55c37dc682ef30629e1b6509e5de6
3
+ size 3378
logs/dreambooth-lora-sd-15/1726467319.0805333/hparams.yml ADDED
@@ -0,0 +1,74 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ adam_beta1: 0.9
2
+ adam_beta2: 0.999
3
+ adam_epsilon: 1.0e-08
4
+ adam_weight_decay: 0.0001
5
+ adam_weight_decay_text_encoder: null
6
+ allow_tf32: false
7
+ cache_dir: null
8
+ cache_latents: false
9
+ caption_column: null
10
+ center_crop: false
11
+ checkpointing_steps: 2000
12
+ checkpoints_total_limit: null
13
+ class_data_dir: null
14
+ class_prompt: null
15
+ dataloader_num_workers: 0
16
+ dataset_config_name: null
17
+ dataset_name: huggan/pokemon
18
+ enable_xformers_memory_efficient_attention: false
19
+ gradient_accumulation_steps: 1
20
+ gradient_checkpointing: true
21
+ hub_model_id: null
22
+ hub_token: null
23
+ image_column: image
24
+ instance_data_dir: null
25
+ instance_prompt: <pokemon>
26
+ learning_rate: 1.0
27
+ local_rank: -1
28
+ logging_dir: logs
29
+ lr_num_cycles: 1
30
+ lr_power: 1.0
31
+ lr_scheduler: constant
32
+ lr_warmup_steps: 0
33
+ max_grad_norm: 1.0
34
+ max_train_steps: 1000
35
+ mixed_precision: fp16
36
+ noise_offset: 0
37
+ num_class_images: 100
38
+ num_new_tokens_per_abstraction: 2
39
+ num_train_epochs: 1
40
+ num_validation_images: 4
41
+ optimizer: prodigy
42
+ output_dir: sdpokemon
43
+ pretrained_model_name_or_path: emilianJR/epiCRealism
44
+ pretrained_vae_model_name_or_path: null
45
+ prior_generation_precision: null
46
+ prior_loss_weight: 1.0
47
+ prodigy_beta3: null
48
+ prodigy_decouple: true
49
+ prodigy_safeguard_warmup: true
50
+ prodigy_use_bias_correction: true
51
+ push_to_hub: true
52
+ rank: 8
53
+ repeats: 1
54
+ report_to: tensorboard
55
+ resolution: 512
56
+ resume_from_checkpoint: null
57
+ revision: null
58
+ sample_batch_size: 4
59
+ scale_lr: false
60
+ seed: 0
61
+ snr_gamma: 5.0
62
+ text_encoder_lr: 1.0
63
+ token_abstraction: TOK
64
+ train_batch_size: 1
65
+ train_text_encoder: false
66
+ train_text_encoder_frac: 1.0
67
+ train_text_encoder_ti: true
68
+ train_text_encoder_ti_frac: 0.5
69
+ use_8bit_adam: false
70
+ use_dora: false
71
+ validation_epochs: 50
72
+ validation_prompt: <pokemon> with white background
73
+ variant: null
74
+ with_prior_preservation: false
logs/dreambooth-lora-sd-15/events.out.tfevents.1726467319.dc44792b2f5e.1319.0 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:b59242b410459bf6e084acb86b3d85ec4d4e1c99486ed1ccef245143046c642c
3
+ size 2406266
pytorch_lora_weights.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:31d582a1be5ded44158b2832286877fd0448f405c55705afe78d9c930eaa36ca
3
+ size 6414992
sdpokemon.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:96a5268b4dabe6044c52755ba10f888b2271c1af83458002a035801635e36720
3
+ size 6433712
sdpokemon_emb.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:b248c9cb6c8d0ef01a937b30d77d38902ed4c26c39e84c468f36c5a151798208
3
+ size 3152