HaileyStorm commited on
Commit
ea603a6
1 Parent(s): a4e5856

Upload folder using huggingface_hub

Browse files
Files changed (3) hide show
  1. README.md +46 -0
  2. config.yaml +69 -0
  3. lora.safetensors +3 -0
README.md ADDED
@@ -0,0 +1,46 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ license: other
3
+ license_name: flux-1-dev-non-commercial-license
4
+ license_link: https://huggingface.co/black-forest-labs/FLUX.1-dev/blob/main/LICENSE.md
5
+ language:
6
+ - en
7
+ tags:
8
+ - flux
9
+ - diffusers
10
+ - lora
11
+ - replicate
12
+ base_model: "black-forest-labs/FLUX.1-dev"
13
+ pipeline_tag: text-to-image
14
+ # widget:
15
+ # - text: >-
16
+ # prompt
17
+ # output:
18
+ # url: https://...
19
+ instance_prompt: GRAYSONCO
20
+ ---
21
+
22
+ # Grayson
23
+
24
+ <!-- <Gallery /> -->
25
+
26
+ Trained on Replicate using:
27
+
28
+ https://replicate.com/ostris/flux-dev-lora-trainer/train
29
+
30
+
31
+ ## Trigger words
32
+ You should use `GRAYSONCO` to trigger the image generation.
33
+
34
+
35
+ ## Use it with the [🧨 diffusers library](https://github.com/huggingface/diffusers)
36
+
37
+ ```py
38
+ from diffusers import AutoPipelineForText2Image
39
+ import torch
40
+
41
+ pipeline = AutoPipelineForText2Image.from_pretrained('black-forest-labs/FLUX.1-dev', torch_dtype=torch.float16).to('cuda')
42
+ pipeline.load_lora_weights('HaileyStorm/Grayson', weight_name='lora.safetensors')
43
+ image = pipeline('your prompt').images[0]
44
+ ```
45
+
46
+ 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)
config.yaml ADDED
@@ -0,0 +1,69 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ job: custom_job
2
+ config:
3
+ name: flux_train_replicate
4
+ process:
5
+ - type: custom_sd_trainer
6
+ training_folder: output
7
+ device: cuda:0
8
+ trigger_word: GRAYSONCO
9
+ network:
10
+ type: lora
11
+ linear: 16
12
+ linear_alpha: 16
13
+ network_kwargs:
14
+ only_if_contains:
15
+ - transformer.single_transformer_blocks.7.proj_out
16
+ - transformer.single_transformer_blocks.12.proj_out
17
+ - transformer.single_transformer_blocks.16.proj_out
18
+ - transformer.single_transformer_blocks.20.proj_out
19
+ save:
20
+ dtype: float16
21
+ save_every: 250
22
+ max_step_saves_to_keep: 1
23
+ datasets:
24
+ - folder_path: input_images
25
+ caption_ext: txt
26
+ caption_dropout_rate: 0.05
27
+ shuffle_tokens: false
28
+ cache_latents_to_disk: false
29
+ cache_latents: true
30
+ resolution:
31
+ - 512
32
+ - 768
33
+ - 1024
34
+ train:
35
+ batch_size: 1
36
+ steps: 1500
37
+ gradient_accumulation_steps: 1
38
+ train_unet: true
39
+ train_text_encoder: false
40
+ content_or_style: balanced
41
+ gradient_checkpointing: true
42
+ noise_scheduler: flowmatch
43
+ optimizer: adamw8bit
44
+ lr: 0.0001
45
+ ema_config:
46
+ use_ema: true
47
+ ema_decay: 0.99
48
+ dtype: bf16
49
+ model:
50
+ name_or_path: FLUX.1-dev
51
+ is_flux: true
52
+ quantize: true
53
+ sample:
54
+ sampler: flowmatch
55
+ sample_every: 250
56
+ width: 1024
57
+ height: 1024
58
+ prompts:
59
+ - Portrait of GRAYSONCO dressed as Voldemort for halloween
60
+ - A photo of GRAYSONCO on the moon
61
+ - GRAYSONCO driving a forklift
62
+ neg: ''
63
+ seed: 42
64
+ walk_seed: true
65
+ guidance_scale: 3.5
66
+ sample_steps: 28
67
+ meta:
68
+ name: flux_train_replicate
69
+ version: '1.0'
lora.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:8455665d6cb930390a838bb409ea4de6b364c1adafe263f0a60d6da00c43d5cd
3
+ size 2360856