Model card auto-generated by SimpleTuner

README.md

@@ -1,9 +1,9 @@
 ---
-license: creativeml-openrail-m
-base_model: "stabilityai/stable-diffusion-3-medium-diffusers"
+license: other
+base_model: "black-forest-labs/FLUX.1-schnell"
 tags:
-  - stable-diffusion
-  - stable-diffusion-diffusers
+  - flux
+  - flux-diffusers
   - text-to-image
   - diffusers
   - simpletuner
@@ -16,7 +16,7 @@ widget:
     negative_prompt: 'blurry, cropped, ugly, 3d, colorful'
   output:
     url: ./assets/image_0_0.png
-- text: 'geometric diagram of a right-angle triangle with base 3cm and height 4cm'
+- text: 'A simple, clean line drawing of a right-angled triangle. The right angle is at the bottom left corner. The base is labeled as ''6 cm'' and the height is labeled as ''4 cm''. The drawing is set against a plain white background.'
   parameters:
     negative_prompt: 'blurry, cropped, ugly, 3d, colorful'
   output:
@@ -25,8 +25,7 @@ widget:
 
 # lora-training
 
-This is a LoRA derived from [stabilityai/stable-diffusion-3-medium-diffusers](https://huggingface.co/stabilityai/stable-diffusion-3-medium-diffusers).
-
+This is a standard PEFT LoRA derived from [black-forest-labs/FLUX.1-schnell](https://huggingface.co/black-forest-labs/FLUX.1-schnell).
 
 
 The main validation prompt used during training was:
@@ -34,13 +33,13 @@ The main validation prompt used during training was:
 
 
 ```
-geometric diagram of a right-angle triangle with base 3cm and height 4cm
+A simple, clean line drawing of a right-angled triangle. The right angle is at the bottom left corner. The base is labeled as '6 cm' and the height is labeled as '4 cm'. The drawing is set against a plain white background.
 ```
 
 ## Validation settings
-- CFG: `3.0`
+- CFG: `3.5`
 - CFG Rescale: `0.0`
-- Steps: `30`
+- Steps: `24`
 - Sampler: `None`
 - Seed: `42`
 - Resolution: `512`
@@ -58,9 +57,9 @@ You may reuse the base model text encoder for inference.
 
 ## Training settings
 
-- Training epochs: 13
-- Training steps: 5000
-- Learning rate: 8e-05
+- Training epochs: 0
+- Training steps: 100
+- Learning rate: 0.0008
 - Effective batch size: 1
   - Micro-batch size: 1
   - Gradient accumulation steps: 1
@@ -69,19 +68,19 @@ You may reuse the base model text encoder for inference.
 - Rescaled betas zero SNR: False
 - Optimizer: adamw_bf16
 - Precision: bf16
-- Quantised: No
+- Quantised: Yes: int8-quanto
 - Xformers: Not used
 - LoRA Rank: 16
 - LoRA Alpha: None
 - LoRA Dropout: 0.1
 - LoRA initialisation style: default
-
+    
 
 ## Datasets
 
 ### right-triangles
 - Repeats: 0
-- Total number of images: 380
+- Total number of images: 348
 - Total number of aspect buckets: 1
 - Resolution: 512 px
 - Cropped: True
@@ -96,23 +95,21 @@ You may reuse the base model text encoder for inference.
 import torch
 from diffusers import DiffusionPipeline
 
-model_id = 'stabilityai/stable-diffusion-3-medium-diffusers'
+model_id = 'black-forest-labs/FLUX.1-schnell'
 adapter_id = 'Mujeeb603/lora-training'
 pipeline = DiffusionPipeline.from_pretrained(model_id)
 pipeline.load_lora_weights(adapter_id)
 
-prompt = "geometric diagram of a right-angle triangle with base 3cm and height 4cm"
-negative_prompt = 'blurry, cropped, ugly, 3d, colorful'
+prompt = "A simple, clean line drawing of a right-angled triangle. The right angle is at the bottom left corner. The base is labeled as '6 cm' and the height is labeled as '4 cm'. The drawing is set against a plain white background."
 
 pipeline.to('cuda' if torch.cuda.is_available() else 'mps' if torch.backends.mps.is_available() else 'cpu')
 image = pipeline(
     prompt=prompt,
-    negative_prompt=negative_prompt,
-    num_inference_steps=30,
+    num_inference_steps=24,
     generator=torch.Generator(device='cuda' if torch.cuda.is_available() else 'mps' if torch.backends.mps.is_available() else 'cpu').manual_seed(1641421826),
     width=512,
     height=512,
-    guidance_scale=3.0,
+    guidance_scale=3.5,
 ).images[0]
 image.save("output.png", format="PNG")
 ```