End of training

.gitattributes

@@ -33,3 +33,7 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
 *.zip filter=lfs diff=lfs merge=lfs -text
 *.zst filter=lfs diff=lfs merge=lfs -text
 *tfevents* filter=lfs diff=lfs merge=lfs -text
+image_0.png filter=lfs diff=lfs merge=lfs -text
+image_1.png filter=lfs diff=lfs merge=lfs -text
+image_2.png filter=lfs diff=lfs merge=lfs -text
+image_3.png filter=lfs diff=lfs merge=lfs -text

README.md

@@ -0,0 +1,100 @@
+---
+base_model: HiDream-ai/HiDream-I1-Full
+library_name: diffusers
+license: mit
+instance_prompt: TOK
+widget:
+- text: a woman riding an orca while waving hello and working on her laptop in the
+    style of TOK
+  output:
+    url: image_0.png
+- text: a woman riding an orca while waving hello and working on her laptop in the
+    style of TOK
+  output:
+    url: image_1.png
+- text: a woman riding an orca while waving hello and working on her laptop in the
+    style of TOK
+  output:
+    url: image_2.png
+- text: a woman riding an orca while waving hello and working on her laptop in the
+    style of TOK
+  output:
+    url: image_3.png
+tags:
+- text-to-image
+- diffusers-training
+- diffusers
+- lora
+- hidream
+- hidream-diffusers
+- template:sd-lora
+---
+
+<!-- This model card has been generated automatically according to the information the training script had access to. You
+should probably proofread and complete it, then remove this comment. -->
+
+
+# HiDream Image DreamBooth LoRA - KristjanRRR/HiDream-I1-Full-ink-drawing-lora-4
+
+<Gallery />
+
+## Model description
+
+These are KristjanRRR/HiDream-I1-Full-ink-drawing-lora-4 DreamBooth LoRA weights for HiDream-ai/HiDream-I1-Full.
+
+The weights were trained using [DreamBooth](https://dreambooth.github.io/) with the [HiDream Image diffusers trainer](https://github.com/huggingface/diffusers/blob/main/examples/dreambooth/README_hidream.md).
+
+## Trigger words
+
+You should use `TOK` to trigger the image generation.
+
+## Download model
+
+[Download the *.safetensors LoRA](KristjanRRR/HiDream-I1-Full-ink-drawing-lora-4/tree/main) in the Files & versions tab.
+
+## Use it with the [🧨 diffusers library](https://github.com/huggingface/diffusers)
+
+```py
+    >>> import torch
+    >>> from transformers import PreTrainedTokenizerFast, LlamaForCausalLM
+    >>> from diffusers import HiDreamImagePipeline
+
+    >>> tokenizer_4 = PreTrainedTokenizerFast.from_pretrained("meta-llama/Meta-Llama-3.1-8B-Instruct")
+    >>> text_encoder_4 = LlamaForCausalLM.from_pretrained(
+    ...     "meta-llama/Meta-Llama-3.1-8B-Instruct",
+    ...     output_hidden_states=True,
+    ...     output_attentions=True,
+    ...     torch_dtype=torch.bfloat16,
+    ... )
+
+    >>> pipe = HiDreamImagePipeline.from_pretrained(
+    ...     "HiDream-ai/HiDream-I1-Full",
+    ...     tokenizer_4=tokenizer_4,
+    ...     text_encoder_4=text_encoder_4,
+    ...     torch_dtype=torch.bfloat16,
+    ... )
+    >>> pipe.enable_model_cpu_offload()
+    >>> pipe.load_lora_weights(f"KristjanRRR/HiDream-I1-Full-ink-drawing-lora-4")
+    >>> image = pipe(f"TOK").images[0]
+
+
+```
+
+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)
+
+
+## Intended uses & limitations
+
+#### How to use
+
+```python
+# TODO: add an example code snippet for running this diffusion pipeline
+```
+
+#### Limitations and bias
+
+[TODO: provide examples of latent issues and potential remediations]
+
+## Training details
+
+[TODO: describe the data used to train the model]

logs/dreambooth-hidream-lora/1748782938.569996/events.out.tfevents.1748782938.modal.22.1

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:c97582a9ee81bf2812741bd1fb924c6d32d2ba82b9951a9055018ec78a33ba1b
+size 3569

logs/dreambooth-hidream-lora/1748782938.5723155/hparams.yml

@@ -0,0 +1,76 @@
+adam_beta1: 0.9
+adam_beta2: 0.999
+adam_epsilon: 1.0e-08
+adam_weight_decay: 0.0001
+allow_tf32: false
+bnb_quantization_config_path: null
+cache_dir: null
+cache_latents: false
+caption_column: prompt
+center_crop: false
+checkpointing_steps: 500
+checkpoints_total_limit: null
+class_data_dir: null
+class_prompt: null
+dataloader_num_workers: 0
+dataset_config_name: null
+dataset_name: KristjanRRR/ink-drawing-2
+final_validation_prompt: null
+gradient_accumulation_steps: 1
+gradient_checkpointing: false
+hub_model_id: null
+hub_token: null
+image_column: image
+instance_data_dir: null
+instance_prompt: TOK
+learning_rate: 0.0004
+local_rank: -1
+logging_dir: logs
+logit_mean: 0.0
+logit_std: 1.0
+lora_dropout: 0.0
+lora_layers: null
+lr_num_cycles: 1
+lr_power: 1.0
+lr_scheduler: constant
+lr_warmup_steps: 500
+max_grad_norm: 1.0
+max_sequence_length: 128
+max_train_steps: 440
+mixed_precision: bf16
+mode_scale: 1.29
+num_class_images: 100
+num_train_epochs: 11
+num_validation_images: 4
+offload: false
+optimizer: AdamW
+output_dir: HiDream-I1-Full-ink-drawing-lora-4
+pretrained_model_name_or_path: HiDream-ai/HiDream-I1-Full
+pretrained_text_encoder_4_name_or_path: meta-llama/Meta-Llama-3.1-8B-Instruct
+pretrained_tokenizer_4_name_or_path: meta-llama/Meta-Llama-3.1-8B-Instruct
+prior_loss_weight: 1.0
+prodigy_beta3: null
+prodigy_decouple: true
+prodigy_safeguard_warmup: true
+prodigy_use_bias_correction: true
+push_to_hub: true
+random_flip: false
+rank: 16
+repeats: 2
+report_to: tensorboard
+resolution: 1024
+resume_from_checkpoint: null
+revision: null
+sample_batch_size: 4
+scale_lr: false
+seed: 1
+skip_final_inference: false
+train_batch_size: 1
+upcast_before_saving: false
+use_8bit_adam: false
+validation_epochs: 50
+validation_prompt: a woman riding an orca while waving hello and working on her laptop
+  in the style of TOK
+variant: null
+weighting_scheme: none
+with_prior_preservation: false

logs/dreambooth-hidream-lora/events.out.tfevents.1748782938.modal.22.0

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:83b70f64484f904ac2361d5f04084880ce294b15a3b11bb84f86f046ee6fa886
+size 3705476

pytorch_lora_weights.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:84818098b5685a1be3f78454c0670a8370dca1fd9d833a04d23c140ca30602a8
+size 31510456