End of training

.gitattributes

@@ -33,3 +33,5 @@ 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_1.png filter=lfs diff=lfs merge=lfs -text
+image_2.png filter=lfs diff=lfs merge=lfs -text

README.md

@@ -0,0 +1,96 @@
+---
+tags:
+- stable-diffusion-xl
+- stable-diffusion-xl-diffusers
+- diffusers-training
+- text-to-image
+- diffusers
+- lora
+- template:sd-lora
+widget:
+
+        - text: 'a website screenshot <s0><s1>'
+          output:
+            url:
+                "image_0.png"
+        
+        - text: 'a website screenshot <s0><s1>'
+          output:
+            url:
+                "image_1.png"
+        
+        - text: 'a website screenshot <s0><s1>'
+          output:
+            url:
+                "image_2.png"
+        
+        - text: 'a website screenshot <s0><s1>'
+          output:
+            url:
+                "image_3.png"
+        
+base_model: stabilityai/stable-diffusion-xl-base-1.0
+instance_prompt: a website that looks like <s0><s1>
+license: openrail++
+---
+
+# SDXL LoRA DreamBooth - anant9/website-ui-sdxl-lora
+
+<Gallery />
+
+## Model description
+
+### These are anant9/website-ui-sdxl-lora LoRA adaption weights for stabilityai/stable-diffusion-xl-base-1.0.
+
+## Download model
+
+### Use it with UIs such as AUTOMATIC1111, Comfy UI, SD.Next, Invoke
+
+- **LoRA**: download **[`website-ui-sdxl-lora.safetensors` here 💾](/anant9/website-ui-sdxl-lora/blob/main/website-ui-sdxl-lora.safetensors)**.
+    - Place it on your `models/Lora` folder.
+    - On AUTOMATIC1111, load the LoRA by adding `<lora:website-ui-sdxl-lora:1>` to your prompt. On ComfyUI just [load it as a regular LoRA](https://comfyanonymous.github.io/ComfyUI_examples/lora/).
+- *Embeddings*: download **[`website-ui-sdxl-lora_emb.safetensors` here 💾](/anant9/website-ui-sdxl-lora/blob/main/website-ui-sdxl-lora_emb.safetensors)**.
+    - Place it on it on your `embeddings` folder
+    - Use it by adding `website-ui-sdxl-lora_emb` to your prompt. For example, `a website that looks like website-ui-sdxl-lora_emb`
+    (you need both the LoRA and the embeddings as they were trained together for this LoRA)
+    
+
+## Use it with the [🧨 diffusers library](https://github.com/huggingface/diffusers)
+
+```py
+from diffusers import AutoPipelineForText2Image
+import torch
+from huggingface_hub import hf_hub_download
+from safetensors.torch import load_file
+        
+pipeline = AutoPipelineForText2Image.from_pretrained('stabilityai/stable-diffusion-xl-base-1.0', torch_dtype=torch.float16).to('cuda')
+pipeline.load_lora_weights('anant9/website-ui-sdxl-lora', weight_name='pytorch_lora_weights.safetensors')
+embedding_path = hf_hub_download(repo_id='anant9/website-ui-sdxl-lora', filename='website-ui-sdxl-lora_emb.safetensors', repo_type="model")
+state_dict = load_file(embedding_path)
+pipeline.load_textual_inversion(state_dict["clip_l"], token=["<s0>", "<s1>"], text_encoder=pipeline.text_encoder, tokenizer=pipeline.tokenizer)
+pipeline.load_textual_inversion(state_dict["clip_g"], token=["<s0>", "<s1>"], text_encoder=pipeline.text_encoder_2, tokenizer=pipeline.tokenizer_2)
+        
+image = pipeline('a website screenshot <s0><s1>').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)
+
+## Trigger words
+
+To trigger image generation of trained concept(or concepts) replace each concept identifier in you prompt with the new inserted tokens:
+
+to trigger concept `TOK` → use `<s0><s1>` in your prompt 
+
+
+
+## Details
+All [Files & versions](/anant9/website-ui-sdxl-lora/tree/main).
+
+The weights were trained using [🧨 diffusers Advanced Dreambooth Training Script](https://github.com/huggingface/diffusers/blob/main/examples/advanced_diffusion_training/train_dreambooth_lora_sdxl_advanced.py).
+
+LoRA for the text encoder was enabled. False.
+
+Pivotal tuning was enabled: True.
+
+Special VAE used for training: madebyollin/sdxl-vae-fp16-fix.
+

logs/dreambooth-lora-sd-xl/1712699205.7232225/events.out.tfevents.1712699205.5c78f655cd09.7888.1

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

logs/dreambooth-lora-sd-xl/1712699205.7252805/hparams.yml

@@ -0,0 +1,76 @@
+adam_beta1: 0.9
+adam_beta2: 0.99
+adam_epsilon: 1.0e-08
+adam_weight_decay: 0.0001
+adam_weight_decay_text_encoder: null
+allow_tf32: false
+cache_dir: null
+cache_latents: false
+caption_column: prompt
+center_crop: false
+checkpointing_steps: 2000
+checkpoints_total_limit: null
+class_data_dir: null
+class_prompt: null
+dataloader_num_workers: 0
+dataset_config_name: null
+dataset_name: ./my_folder
+do_edm_style_training: false
+enable_xformers_memory_efficient_attention: false
+gradient_accumulation_steps: 1
+gradient_checkpointing: true
+hub_model_id: null
+hub_token: null
+image_column: image
+instance_data_dir: null
+instance_prompt: a website that looks like <s0><s1>
+learning_rate: 1.0
+local_rank: -1
+logging_dir: logs
+lr_num_cycles: 1
+lr_power: 1.0
+lr_scheduler: constant
+lr_warmup_steps: 0
+max_grad_norm: 1.0
+max_train_steps: 1000
+mixed_precision: bf16
+noise_offset: 0
+num_class_images: 100
+num_new_tokens_per_abstraction: 2
+num_train_epochs: 30
+num_validation_images: 4
+optimizer: prodigy
+output_dir: website-ui-sdxl-lora
+pretrained_model_name_or_path: stabilityai/stable-diffusion-xl-base-1.0
+pretrained_vae_model_name_or_path: madebyollin/sdxl-vae-fp16-fix
+prior_generation_precision: null
+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: 8
+repeats: 1
+report_to: tensorboard
+resolution: 1024
+resume_from_checkpoint: null
+revision: null
+sample_batch_size: 4
+scale_lr: false
+seed: 0
+snr_gamma: 5.0
+text_encoder_lr: 1.0
+token_abstraction: TOK
+train_batch_size: 3
+train_text_encoder: false
+train_text_encoder_frac: 1.0
+train_text_encoder_ti: true
+train_text_encoder_ti_frac: 0.5
+use_8bit_adam: false
+use_dora: false
+validation_epochs: 50
+validation_prompt: a website screenshot <s0><s1>
+variant: null
+with_prior_preservation: false

logs/dreambooth-lora-sd-xl/events.out.tfevents.1712699205.5c78f655cd09.7888.0

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

pytorch_lora_weights.safetensors

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

website-ui-sdxl-lora.safetensors

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

website-ui-sdxl-lora_emb.safetensors

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