Upload folder using huggingface_hub

README.md

@@ -16,7 +16,7 @@ pipeline_tag: text-to-image
 #       prompt
 #     output:
 #       url: https://...
-instance_prompt: Puchi
+instance_prompt: sk
 ---
 
 # Puchi
@@ -32,7 +32,7 @@ It was trained on [Replicate](https://replicate.com/) using AI toolkit: https://
 
 ## Trigger words
 
-You should use `Puchi` to trigger the image generation.
+You should use `sk` to trigger the image generation.
 
 
 ## Run this LoRA with an API using Replicate
@@ -41,7 +41,7 @@ You should use `Puchi` to trigger the image generation.
 import replicate
 
 input = {
-    "prompt": "Puchi",
+    "prompt": "sk",
     "lora_weights": "https://huggingface.co/Ankitvaibs/Puchi/resolve/main/lora.safetensors"
 }
 
@@ -62,7 +62,7 @@ import torch
 
 pipeline = AutoPipelineForText2Image.from_pretrained('black-forest-labs/FLUX.1-dev', torch_dtype=torch.float16).to('cuda')
 pipeline.load_lora_weights('Ankitvaibs/Puchi', weight_name='lora.safetensors')
-image = pipeline('Puchi').images[0]
+image = pipeline('sk').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)
@@ -70,9 +70,9 @@ For more details, including weighting, merging and fusing LoRAs, check the [docu
 
 ## Training details
 
-- Steps: 4110
+- Steps: 1763
 - Learning rate: 0.0004
-- LoRA rank: 80
+- LoRA rank: 16
 
 
 ## Contribute your own examples

config.yaml

@@ -5,14 +5,14 @@ config:
   - type: custom_sd_trainer
     training_folder: output
     device: cuda:0
-    trigger_word: Puchi
+    trigger_word: sk
     network:
       type: lora
-      linear: 80
-      linear_alpha: 80
+      linear: 16
+      linear_alpha: 16
     save:
       dtype: float16
-      save_every: 4111
+      save_every: 1764
       max_step_saves_to_keep: 1
     datasets:
     - folder_path: input_images
@@ -27,7 +27,7 @@ config:
       - 1024
     train:
       batch_size: 1
-      steps: 4110
+      steps: 1763
       gradient_accumulation_steps: 1
       train_unet: true
       train_text_encoder: false
@@ -46,7 +46,7 @@ config:
       quantize: false
     sample:
       sampler: flowmatch
-      sample_every: 4111
+      sample_every: 1764
       width: 1024
       height: 1024
       prompts: []

lora.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:67a15c3a80aa6279c4439a9f8bd611ad6c457bf12d95b3a7de1c6d22a844683b
-size 859312032
+oid sha256:4bc2364b6a9f8c8d93d73eaa1407d23628edcacbdb240a0d56101524eacbf471
+size 171969408