nerijs commited on
Commit
a7e97c4
1 Parent(s): c9620cf

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -105
README.md CHANGED
@@ -1,105 +1 @@
1
- ---
2
- license: creativeml-openrail-m
3
- base_model: "black-forest-labs/FLUX.1-dev"
4
- tags:
5
- - stable-diffusion
6
- - stable-diffusion-diffusers
7
- - text-to-image
8
- - diffusers
9
- - simpletuner
10
- - lora
11
- - template:sd-lora
12
- inference: true
13
-
14
- ---
15
-
16
- # AlbertFLUX
17
-
18
- This is a LoRA derived from [black-forest-labs/FLUX.1-dev](https://huggingface.co/black-forest-labs/FLUX.1-dev).
19
-
20
-
21
-
22
- The main validation prompt used during training was:
23
-
24
- ```
25
- Albert the cat riding a horse on the moon
26
- ```
27
-
28
- ## Validation settings
29
- - CFG: `3.5`
30
- - CFG Rescale: `0.0`
31
- - Steps: `30`
32
- - Sampler: `None`
33
- - Seed: `42`
34
- - Resolution: `1024`
35
-
36
- Note: The validation settings are not necessarily the same as the [training settings](#training-settings).
37
-
38
-
39
-
40
-
41
- <Gallery />
42
-
43
- The text encoder **was not** trained.
44
- You may reuse the base model text encoder for inference.
45
-
46
-
47
- ## Training settings
48
-
49
- - Training epochs: 55
50
- - Training steps: 1400
51
- - Learning rate: 1.0
52
- - Effective batch size: 1
53
- - Micro-batch size: 1
54
- - Gradient accumulation steps: 1
55
- - Number of GPUs: 1
56
- - Prediction type: epsilon
57
- - Rescaled betas zero SNR: False
58
- - Optimizer: Prodigy
59
- - Precision: bf16
60
- - Xformers: Not used
61
- - LoRA Rank: 16
62
- - LoRA Alpha: 16
63
- - LoRA Dropout: 0.1
64
- - LoRA initialisation style: default
65
-
66
-
67
- ## Datasets
68
-
69
- ### Albert
70
- - Repeats: 0
71
- - Total number of images: 25
72
- - Total number of aspect buckets: 1
73
- - Resolution: 1 megapixels
74
- - Cropped: True
75
- - Crop style: center
76
- - Crop aspect: square
77
-
78
-
79
- ## Inference
80
-
81
-
82
- ```python
83
- import torch
84
- from diffusers import DiffusionPipeline
85
-
86
- model_id = 'black-forest-labs/FLUX.1-dev'
87
- adapter_id = 'AlbertFLUX'
88
- pipeline = DiffusionPipeline.from_pretrained(model_id)\pipeline.load_lora_weights(adapter_id)
89
-
90
- prompt = "Albert the cat riding a horse on the moon"
91
- negative_prompt = "blurry, cropped, ugly"
92
-
93
- pipeline.to('cuda' if torch.cuda.is_available() else 'mps' if torch.backends.mps.is_available() else 'cpu')
94
- image = pipeline(
95
- prompt=prompt,
96
- negative_prompt='blurry, cropped, ugly',
97
- num_inference_steps=30,
98
- generator=torch.Generator(device='cuda' if torch.cuda.is_available() else 'mps' if torch.backends.mps.is_available() else 'cpu').manual_seed(1641421826),
99
- width=1152,
100
- height=768,
101
- guidance_scale=3.5,
102
- guidance_rescale=0.0,
103
- ).images[0]
104
- image.save("output.png", format="PNG")
105
- ```
 
1
+ a collection of really sus loras...