Upload folder using huggingface_hub
Browse files- README.md +9 -2
- config.yaml +8 -7
- lora.safetensors +1 -1
README.md
CHANGED
|
@@ -11,18 +11,25 @@ tags:
|
|
| 11 |
- replicate
|
| 12 |
base_model: "black-forest-labs/FLUX.1-dev"
|
| 13 |
pipeline_tag: text-to-image
|
| 14 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 15 |
---
|
| 16 |
|
| 17 |
# Newmod
|
| 18 |
|
|
|
|
|
|
|
| 19 |
Trained on Replicate using:
|
| 20 |
|
| 21 |
https://replicate.com/ostris/flux-dev-lora-trainer/train
|
| 22 |
|
| 23 |
|
| 24 |
## Trigger words
|
| 25 |
-
You should use `
|
| 26 |
|
| 27 |
|
| 28 |
## Use it with the [🧨 diffusers library](https://github.com/huggingface/diffusers)
|
|
|
|
| 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: TOK
|
| 20 |
---
|
| 21 |
|
| 22 |
# Newmod
|
| 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 `TOK` to trigger the image generation.
|
| 33 |
|
| 34 |
|
| 35 |
## Use it with the [🧨 diffusers library](https://github.com/huggingface/diffusers)
|
config.yaml
CHANGED
|
@@ -5,28 +5,29 @@ config:
|
|
| 5 |
- type: custom_sd_trainer
|
| 6 |
training_folder: output
|
| 7 |
device: cuda:0
|
| 8 |
-
trigger_word:
|
| 9 |
network:
|
| 10 |
type: lora
|
| 11 |
linear: 16
|
| 12 |
linear_alpha: 16
|
| 13 |
save:
|
| 14 |
dtype: float16
|
| 15 |
-
save_every:
|
| 16 |
max_step_saves_to_keep: 1
|
| 17 |
datasets:
|
| 18 |
- folder_path: input_images
|
| 19 |
caption_ext: txt
|
| 20 |
caption_dropout_rate: 0.05
|
| 21 |
shuffle_tokens: false
|
| 22 |
-
cache_latents_to_disk:
|
|
|
|
| 23 |
resolution:
|
| 24 |
- 512
|
| 25 |
- 768
|
| 26 |
- 1024
|
| 27 |
train:
|
| 28 |
batch_size: 1
|
| 29 |
-
steps:
|
| 30 |
gradient_accumulation_steps: 1
|
| 31 |
train_unet: true
|
| 32 |
train_text_encoder: false
|
|
@@ -45,15 +46,15 @@ config:
|
|
| 45 |
quantize: true
|
| 46 |
sample:
|
| 47 |
sampler: flowmatch
|
| 48 |
-
sample_every:
|
| 49 |
width: 1024
|
| 50 |
height: 1024
|
| 51 |
prompts: []
|
| 52 |
neg: ''
|
| 53 |
seed: 42
|
| 54 |
walk_seed: true
|
| 55 |
-
guidance_scale:
|
| 56 |
-
sample_steps:
|
| 57 |
meta:
|
| 58 |
name: flux_train_replicate
|
| 59 |
version: '1.0'
|
|
|
|
| 5 |
- type: custom_sd_trainer
|
| 6 |
training_folder: output
|
| 7 |
device: cuda:0
|
| 8 |
+
trigger_word: TOK
|
| 9 |
network:
|
| 10 |
type: lora
|
| 11 |
linear: 16
|
| 12 |
linear_alpha: 16
|
| 13 |
save:
|
| 14 |
dtype: float16
|
| 15 |
+
save_every: 1498
|
| 16 |
max_step_saves_to_keep: 1
|
| 17 |
datasets:
|
| 18 |
- folder_path: input_images
|
| 19 |
caption_ext: txt
|
| 20 |
caption_dropout_rate: 0.05
|
| 21 |
shuffle_tokens: false
|
| 22 |
+
cache_latents_to_disk: false
|
| 23 |
+
cache_latents: true
|
| 24 |
resolution:
|
| 25 |
- 512
|
| 26 |
- 768
|
| 27 |
- 1024
|
| 28 |
train:
|
| 29 |
batch_size: 1
|
| 30 |
+
steps: 1497
|
| 31 |
gradient_accumulation_steps: 1
|
| 32 |
train_unet: true
|
| 33 |
train_text_encoder: false
|
|
|
|
| 46 |
quantize: true
|
| 47 |
sample:
|
| 48 |
sampler: flowmatch
|
| 49 |
+
sample_every: 1498
|
| 50 |
width: 1024
|
| 51 |
height: 1024
|
| 52 |
prompts: []
|
| 53 |
neg: ''
|
| 54 |
seed: 42
|
| 55 |
walk_seed: true
|
| 56 |
+
guidance_scale: 3.5
|
| 57 |
+
sample_steps: 28
|
| 58 |
meta:
|
| 59 |
name: flux_train_replicate
|
| 60 |
version: '1.0'
|
lora.safetensors
CHANGED
|
@@ -1,3 +1,3 @@
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
-
oid sha256:
|
| 3 |
size 171969416
|
|
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:3c2c62cc97561fc15b68040334810b6b1b9ce5fc13f10fb08255cfcdaea091f5
|
| 3 |
size 171969416
|