maulikanalog commited on
Commit
6de2770
·
verified ·
1 Parent(s): 2bbac02

Upload 4 files

Browse files
Files changed (5) hide show
  1. .gitattributes +2 -0
  2. README.md +50 -0
  3. akshat.safetensors +3 -0
  4. config.yaml +65 -0
  5. optimizer.pt +3 -0
.gitattributes ADDED
@@ -0,0 +1,2 @@
 
 
 
1
+ akshat.safetensors filter=lfs diff=lfs merge=lfs -text
2
+ optimizer.pt filter=lfs diff=lfs merge=lfs -text
README.md ADDED
@@ -0,0 +1,50 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ tags:
3
+ - text-to-image
4
+ - flux
5
+ - lora
6
+ - diffusers
7
+ - template:sd-lora
8
+ - ai-toolkit
9
+ widget:
10
+ - text: 'a person sitting in a cafe working on a screen '
11
+ output:
12
+ url: samples/1736594491197__000002000_0.jpg
13
+ - text: 'a person in suite in cafe '
14
+ output:
15
+ url: samples/1736594509686__000002000_1.jpg
16
+ base_model: black-forest-labs/FLUX.1-dev
17
+ instance_prompt: akshatmodel98
18
+ license: other
19
+ license_name: flux-1-dev-non-commercial-license
20
+ license_link: https://huggingface.co/black-forest-labs/FLUX.1-dev/blob/main/LICENSE.md
21
+ ---
22
+
23
+ # akshat
24
+ Model trained with [AI Toolkit by Ostris](https://github.com/ostris/ai-toolkit)
25
+ <Gallery />
26
+
27
+ ## Trigger words
28
+
29
+ You should use `akshatmodel98` to trigger the image generation.
30
+
31
+ ## Download model and use it with ComfyUI, AUTOMATIC1111, SD.Next, Invoke AI, etc.
32
+
33
+ Weights for this model are available in Safetensors format.
34
+
35
+ [Download](/None/tree/main) them in the Files & versions tab.
36
+
37
+ ## Use it with the [🧨 diffusers library](https://github.com/huggingface/diffusers)
38
+
39
+ ```py
40
+ from diffusers import AutoPipelineForText2Image
41
+ import torch
42
+
43
+ pipeline = AutoPipelineForText2Image.from_pretrained('black-forest-labs/FLUX.1-dev', torch_dtype=torch.bfloat16).to('cuda')
44
+ pipeline.load_lora_weights('None', weight_name='akshat')
45
+ image = pipeline('a person sitting in a cafe working on a screen ').images[0]
46
+ image.save("my_image.png")
47
+ ```
48
+
49
+ 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)
50
+
akshat.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:5bd8f973ccaa2d60497c011e4e2b969a4a1da27d6188a719bfff16e3552c2516
3
+ size 171969408
config.yaml ADDED
@@ -0,0 +1,65 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ job: extension
2
+ config:
3
+ name: akshat
4
+ process:
5
+ - type: sd_trainer
6
+ training_folder: output
7
+ device: cuda:0
8
+ network:
9
+ type: lora
10
+ linear: 16
11
+ linear_alpha: 16
12
+ save:
13
+ dtype: float16
14
+ save_every: 10000
15
+ max_step_saves_to_keep: 4
16
+ push_to_hub: true
17
+ hf_private: true
18
+ datasets:
19
+ - folder_path: datasets/92fd4b6e-b71e-402e-8748-c6f427b56808
20
+ caption_ext: txt
21
+ caption_dropout_rate: 0.05
22
+ shuffle_tokens: false
23
+ cache_latents_to_disk: true
24
+ resolution:
25
+ - 512
26
+ - 768
27
+ - 1024
28
+ train:
29
+ batch_size: 1
30
+ steps: 2000
31
+ gradient_accumulation_steps: 1
32
+ train_unet: true
33
+ train_text_encoder: false
34
+ gradient_checkpointing: true
35
+ noise_scheduler: flowmatch
36
+ optimizer: adamw8bit
37
+ lr: 0.0004
38
+ ema_config:
39
+ use_ema: true
40
+ ema_decay: 0.99
41
+ dtype: bf16
42
+ skip_first_sample: true
43
+ disable_sampling: false
44
+ model:
45
+ name_or_path: black-forest-labs/FLUX.1-dev
46
+ is_flux: true
47
+ quantize: true
48
+ low_vram: false
49
+ sample:
50
+ sampler: flowmatch
51
+ sample_every: 1000
52
+ width: 1024
53
+ height: 1024
54
+ prompts:
55
+ - 'a person sitting in a cafe working on a screen '
56
+ - 'a person in suite in cafe '
57
+ neg: ''
58
+ seed: 42
59
+ walk_seed: true
60
+ guidance_scale: 3.5
61
+ sample_steps: 28
62
+ trigger_word: akshatmodel98
63
+ meta:
64
+ name: akshat
65
+ version: '1.0'
optimizer.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:1361d508f44a52548bf27b14ec069c7d38e5492f96e5bb9ac8169d8b166add29
3
+ size 173272836