aimalias commited on
Commit
5f2d880
·
verified ·
1 Parent(s): eaa8cd3

End of training

Browse files
README.md ADDED
@@ -0,0 +1,73 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ base_model: krea/Krea-2-Raw
3
+ library_name: diffusers
4
+ license: apache-2.0
5
+ instance_prompt: TOK
6
+ widget: []
7
+ tags:
8
+ - text-to-image
9
+ - diffusers-training
10
+ - diffusers
11
+ - lora
12
+ - krea2
13
+ - krea2-diffusers
14
+ - template:sd-lora
15
+ ---
16
+
17
+ <!-- This model card has been generated automatically according to the information the training script had access to. You
18
+ should probably proofread and complete it, then remove this comment. -->
19
+
20
+
21
+ # Krea 2 DreamBooth LoRA - aimalias/k4td
22
+
23
+ <Gallery />
24
+
25
+ ## Model description
26
+
27
+ These are aimalias/k4td DreamBooth LoRA weights, trained on krea/Krea-2-Raw.
28
+
29
+ The weights were trained using [DreamBooth](https://dreambooth.github.io/) with the [Krea 2 diffusers trainer](https://github.com/huggingface/diffusers/blob/main/examples/dreambooth/README_krea2.md).
30
+
31
+ Krea 2 ships as two checkpoints: **RAW** (the non-distilled base you fine-tune on) and **Turbo** (an 8-step distilled checkpoint for fast, high-quality inference). Train your LoRA on RAW and run it on Turbo — LoRAs trained on RAW express strongly on Turbo.
32
+
33
+ ## Trigger words
34
+
35
+ You should use `TOK` to trigger the image generation.
36
+
37
+ ## Download model
38
+
39
+ [Download the *.safetensors LoRA](aimalias/k4td/tree/main) in the Files & versions tab.
40
+
41
+ ## Use it with the [🧨 diffusers library](https://github.com/huggingface/diffusers)
42
+
43
+ ```py
44
+ >>> import torch
45
+ >>> from diffusers import Krea2Pipeline
46
+
47
+ >>> # Load the LoRA onto Krea 2 Turbo (the distilled inference model)
48
+ >>> pipe = Krea2Pipeline.from_pretrained("krea/Krea-2-Turbo", torch_dtype=torch.bfloat16).to("cuda")
49
+ >>> pipe.load_lora_weights("aimalias/k4td")
50
+
51
+ >>> # Turbo recipe: 8 steps, no classifier-free guidance
52
+ >>> image = pipe("TOK", num_inference_steps=8, guidance_scale=0.0).images[0]
53
+ >>> image.save("output.png")
54
+ ```
55
+
56
+ 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)
57
+
58
+
59
+ ## Intended uses & limitations
60
+
61
+ #### How to use
62
+
63
+ ```python
64
+ # TODO: add an example code snippet for running this diffusion pipeline
65
+ ```
66
+
67
+ #### Limitations and bias
68
+
69
+ [TODO: provide examples of latent issues and potential remediations]
70
+
71
+ ## Training details
72
+
73
+ [TODO: describe the data used to train the model]
checkpoint-1000/optimizer.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:c84e511a2bdccadeddcd5ef107942cdc2a0826336d6f0bc00153a68a139b344d
3
+ size 195389021
checkpoint-1000/pytorch_lora_weights.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:078fb344cb7be74eb74753e0b8f1805843b2a6a0dec1ae5bc332cfb1ab978e26
3
+ size 191879256
checkpoint-1000/random_states_0.pkl ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:ccd704943a0b21a7331237d378889d5c860b5ebfc39b22dd8fd7e699739a8c5f
3
+ size 14757
checkpoint-1000/scheduler.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:95d5da8195335d4ea322bbf2d1a3e2e6db077cf77404469c5cb95a451f7e56fa
3
+ size 1401
checkpoint-500/optimizer.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:3ef027d2b1b381463d0670e1520a652e510a3eac125a15fa3654c69665cd6529
3
+ size 195389021
checkpoint-500/pytorch_lora_weights.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:5e4d3b8123515c2084b0593ccf330c59c132d320469e7bc07ccd3ad16dc836bc
3
+ size 191879256
checkpoint-500/random_states_0.pkl ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:7bc8c78344bae5513f2ccfe8fdd5a63175c36a853bd66680bc18855bd72152b0
3
+ size 14757
checkpoint-500/scheduler.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:f6b09a32f9bd8c3811caf53e7088f9d0094eade3af68c322a05eb696814b1a19
3
+ size 1401
pytorch_lora_weights.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:0450059fc748d704cfdd29aa137f4eef30b3e7bfd15ceff153d8fc3d63628b90
3
+ size 191879256